使用 pytorch 展开后图像会变色

Question

我是 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])

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

下面是问题的截图

上为贴片，下为全图

任何帮助表示赞赏，谢谢

Answer 1

我认为您的数据集可能很好。

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

如果您可以分享您用于 plot matplotlib 图的完整代码，将会很有帮助。

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

如果是这种情况，我建议您尝试 plot 这个

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