使用 pytorch 展开后图像会改变颜色答案

【问题标题】：Images change color after unfolding using pytorch使用 pytorch 展开后图像会改变颜色
【发布时间】：2020-03-21 04:29:12
【问题描述】：

我是 pytorch 的新手，对于我的一个项目，我将大图像分割成更小的图块/补丁。我正在使用展开来实现这一点。我的代码如下

data = training_set[1][0].data.unfold(1, 64, 64).unfold(2, 64, 64).unfold(3, 64, 64)

这样做后，由于图像被翻转，所以我转置了结果矩阵，就像这个示例代码一样

torch.t(data [0][0][0][0])

但生成的图像会因某种原因失去颜色或变色，我担心这可能会影响我基于这些补丁所做的任何计算。

以下是问题截图

上面是补丁，下面是全图

感谢任何帮助，谢谢

【问题讨论】：

如果您遵循本教程pytorch.org/tutorials/beginner/data_loading_tutorial.html，您将获得裁剪自定义转换，我建议使用数据加载器应用任何形式的转换，即使您刚刚开始，因为它会有所帮助当你有更多数据时。还有你要绘制的图像有多少通道？

【解决方案1】：

我认为您的数据集可能没问题。

我过去曾遇到过与此相关的类似问题。在这种情况下，我感觉罪魁祸首是 matplotlib.imshow() 函数。

如果您能分享您用来绘制 matplotlib 图的完整代码，将会很有帮助。

您很可能将 RGBA 而不是常规 RGB 作为 plt.imshow() 函数的输入。因此，颜色只是因为您还在常规的红绿蓝 (RGB) 之上显示了一个 alpha 值 (A)。

如果是这样的话，我建议你尝试绘制这个

image = torch.t(data [0][0][0])

【讨论】：