为何PIL与Matplotlib的图像输出存在差异？技术咨询

以下是导致两者输出差异的核心原因：

1. 通道顺序的处理逻辑不同 #

这是你遇到问题的主要原因：

• transforms.ToPILImage()默认将输入的(C, H, W)张量的通道视为RGB顺序来生成图像。但你的图像是通过OpenCV加载的，张量通道为BGR顺序，因此PIL会错误地把BGR当成RGB解析，导致颜色完全错乱（比如红色和蓝色通道互换），这就是输出不符合预期的根源。
• Matplotlib的plt.imshow()虽然默认也期望RGB顺序，但你直接传入了未转换通道的(H, W, C)格式BGR张量，显示出的就是你预期的BGR原始颜色（也就是你提到的偏蓝效果），因此输出符合预期。

2. 图像维度排列的要求不同 #

两个库对输入张量的维度格式要求不一样：

• PIL的ToPILImage()接受(通道数C, 高度H, 宽度W)格式的张量，你的代码中squeeze()后的张量正好满足这个要求，所以维度层面没有问题。
• Matplotlib的plt.imshow()要求输入为(高度H, 宽度W, 通道数C)格式，因此你必须通过permute(1,2,0)完成维度转置，否则会出现维度不匹配导致的显示错误。

额外补充：数据范围的潜在影响 #

如果你的张量是浮点型，两个库对数据范围的要求都是[0, 1]；如果是整型则要求[0, 255]。如果你的张量数据超出对应范围，两个库都会自动截断，但从你的代码来看，这不是本次差异的主要原因。

内容的提问来源于stack exchange，提问作者hasin