python - plt 图像颜色错误

Question

我正在研究 CNN 模型并在图像上进行了测试，输出必须是图像

这是我的代码

im = Image.open('1.jpg')
transform = transforms.Compose([transforms.ToTensor(),])
image = transform(im)
tensor = image.unsqueeze(1)
print(tensor.shape)
output = model(tensor.permute(2, 0, 3, 1))
data_img = (output.squeeze()*255.0)
x=data_img.detach().numpy().astype(np.uint8)
img2=np.swapaxes(x,2,1)
trans = transforms.ToPILImage(mode='RGB')
plt.imshow(trans(img2.squeeze()))
plt.show()

我得到了这张图片

在此处输入图像描述

我应该得到这个

在此处输入图像描述

标签： pythonconv-neural-network