用图像改变 numpy 数组的形状

Question

我有一个名为 FLAIR_X 的 NumPy 数组，其中包含 583 个图像。可以使用索引（例如，plt.imshow(FLAIR_X[0])）提取每个图像。目前，数组的形状是(583,)，但是sklearn需要一个大小为(583, 224, 224)的数组（(224,224)是单个图像的大小）。这也将使其更容易使用。我试着做：

temp = FLAIR_X.reshape(583, 224, 224)

但我得到了错误：

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-108-4bb1899edbd8> in <module>
----> 1 temp = FLAIR_X.reshape(583, 224, 224)

ValueError: cannot reshape array of size 583 into shape (583,224,224)

如何重塑这个数组，使其符合要求？

P.S：只是一个额外的问题，在我重塑数据后，我能否像之前 (plt.imshow(FLAIR_X[0])) 那样显示图像？

谢谢，

Answer 1

您是否尝试过简单地调用：

res = np.array(FLAIR_X)

如果没有帮助，请尝试：

l = FLAIR_X.tolist()
res = np.array(l)

如果这也没有帮助，我不相信你的元素是 numpy 数组。