python - 无法将 numpy 数组 reshape 为 RGB 图像

Question

我有一个 numpy 数组，它包含来自 CIFAR10 数据集的所有图像:(50000, 3072)我从中取出一个数组，它的形状是(3072,)然后我尝试通过执行将其转换为 RGB 图像.reshape(32, 32, 3)但是当我将它传递给 plt.show() 时，它会显示类似于 的内容
我不确定为什么它不能正确显示图像，我也试过了.reshape(3, 32, 32)但它给出了TypeError: Invalid shape (3, 32, 32) for image data在绘图时
如何正确 reshape 它？

Answer 1

你应该试试np.swapaxes()获取 plt.imshow() 的有效形状接受。

import numpy as np
dd = np.array([1]*3072).reshape(3, 32, 32)
dd1 = np.swapaxes(dd, 0, 2)

试试 plt.imshow(dd1)看看会发生什么。 (使用您自己的数据):p
首先，您应该使用 (3, 32, 32) 进行整形，这取决于您的数据顺序。
然后你必须显示 reshape 图像，但是 plt.imshow()只接受形状为 (M, N) 或 (M, N, 3) 或 (M, N, 4) 的数组，因此您现在必须交换轴，因为重新整形的数据是 (3, 32, 32)。详情见 plt.imshow和 here