python - PyTorch: 'ToTensor()' 将彩色图片变成 9 张灰度图片

Question

我发现当我使用'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)

Answer 1

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

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