我试图显示存储为Pytorch张量的图像。

trainset = datasets.ImageFolder('data/Cat_Dog_data/train/', transform=transforms)trainload = torch.utils.data.DataLoader(trainset, batch_size=32, shuffle=True)images, labels = iter(trainload).next()image = images[0]image.shape >>> torch.Size([3, 224, 224]) # pyplot 不喜欢这个格式，所以需要重塑image = image.reshape(224,224,3)plt.imshow(image.numpy())

这种方法显示的是同一张图像的3×3网格，并且总是以灰度显示。例如：

如何修复这个问题，以便正确显示单色图像？

回答：

这很奇怪。尝试通过置换而不是重塑来将通道放在最后：

image.permute(1, 2, 0)