TypeError:画像データのサイズが無効です
Typeerror Invalid Dimensions
動機: 3D出力の視覚化
import torch import torchvision import matplotlib.pyplot as plt image = torch.randn(1, 16, 256, 256)#(NDHW) image = image.permute(1,0,2,3)#torch.Size([16, 1, 256, 256]) #viewed as (NCHW) only for the following visulization img = torchvision.utils.make_grid(image, nrow=4,normalize=True).permute(1, 2, 0) plt.imshow(img) plt.show()
のコードがない場合
.permute(1, 2, 0)
次のエラーが発生する可能性があります TypeError:画像データのサイズが無効です
理由: チャネルがdim2に格納されるように、画像テンソルを並べ替える必要があります。