PyTorch：'ToTensor()' 将彩色图像变成 9 张灰度图片答案

【问题标题】：PyTorch: 'ToTensor()' turns color image into 9 grayscale picturesPyTorch：'ToTensor()' 将彩色图像变成 9 张灰度图片
【发布时间】：2020-10-31 03:09:08
【问题描述】：

我发现当我对一个图像使用'ToTensor'时，一个图像显示为9。我查看了官方文档但找不到原因。那为什么一张图变成9张图？？？问题如下图。

a = plt.imread('test.jpg')
plt.imshow(a)
plt.show()

transform = transforms.Compose([transforms.ToTensor()])
b = transform(a)
b = b.view(375,500,3)
plt.imshow(b)

【问题讨论】：

标签： python machine-learning deep-learning pytorch

【解决方案1】：

当您使用transforms.ToTensor() 时，默认情况下它会将输入数组从 HWC 更改为 CHW 顺序。对于绘图，您需要将通道推回到最后一个维度。

plt.imshow(b.permute(2, 0, 1))

【讨论】：