Pil

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に格納されるように、画像テンソルを並べ替える必要があります。