使用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])

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

下面是问题的截图

上面是补丁，下面是完整的图片

谢谢你的帮助