当显示带有 plt.show() 的图像时 -> MemoryError: Unable to allocate array with shape (3600, 7200, 4) and data type float32

Question

我正在尝试 plot 一个图像，上面绘制了一些东西（如点、轴等）。 为此，我使用库matplotlib.pyplot 。 这是我使用的（简化的）代码：

fig = plt.figure(figsize=(18, 18))
ax = fig.add_subplot(111)
img = plt.imread(stim_dir + image_name)

ax.imshow(img, extent=[-np.pi, np.pi, -np.pi / 2, np.pi / 2])
ax.scatter(*zip(*points), marker='^', s=1, c='red', alpha=0.4)
plt.xlim((-np.pi, np.pi))

plt.title(file_name)
plt.show()

现在，当我使用 png 图像时，代码可以正常工作，一切都很完美。 但是我也有到 plot 的 bmp 文件，但它不起作用，无法读取 bitmap，但不用担心我可以将其转换为 png 文件，但是：这些新的 png 图像也不起作用，我得到错误： MemoryError: Unable to allocate array with shape (3600, 7200, 4) and data type float32 。

因此，我打印了工作正常的初始 png 图像的形状，以查看差异：

>>> print(img.shape)
(2160, 4320, 3)

我的猜测是第三维应该是 3（对于 RGB）而不是 4（可能是 ARGB），所以我用“新”不工作的图像打印 img 数组，这给了我类似的东西：

[[[0.46666667 0.46666667 0.46666667 1.        ]
  [0.46666667 0.46666667 0.46666667 1.        ]
  [0.46666667 0.46666667 0.46666667 1.        ]
  ...

查了一下，发现每4个数就等于1，好整齐，我只需要删掉它，它没有携带重要信息！ 我使用img = np.delete(img, 3, 2)并且效果很好，我得到了(3600, 7200, 3)的新形状以及我想要的每个像素的值。 但是，它仍然不起作用，并且仍然给我与以前相同的错误： MemoryError: Unable to allocate array with shape (3600, 7200, 4) and data type float32 ，似乎它不采用新的 img 数组...（显然我在指令img = plt.imread(stim_dir + image_name)之后执行删除）。

那么，我错过了一些明显的东西吗？ 我该怎么做才能让它发挥作用？

Answer 1

对不起，不是真正的解决方案：

显然 matplotlib 无法将足够的 memory 分配给 plot 你所有的数据点，最后你有 3600*78000*4 中的 3600*7800*4。 不考虑散点图，你也想 plot 。

如果您尝试仅对图像的一部分进行 plot 会发生什么？ 让我们说左上角？