python - 有没有办法将 pyplot.imshow() 对象转换为 numpy 数组？

Question

我需要可视化一个 2D numpy 数组。我为此使用 pyplot。这是代码：

import cv2 as cv
import numpy as np
from matplotlib import pyplot

img = pyplot.imshow( radiance_val )
#radiance_val is a 2D numpy array of size = ( 512, 512 ) 
#filled with np.float32 values

pyplot.show()

我得到了预期的输出。

现在我的问题是，有什么方法可以将上述代码中的“img”从 pyplot 类型转换为 numpy 类型。我需要这个，以便我可以将可视化加载为 opencv 图像并对其进行进一步处理。我正在使用 32 位的 python 2.7。

请帮助

谢谢

---

编辑 1：在 Thorsten Kranz 的解决方案之后

import numpy as np
import cv2 as cv
import matplotlib.pyplot as plt
import PIL
from cStringIO import StringIO

frame1 = plt.gca()
frame1.axes.get_xaxis().set_visible(False)
frame1.axes.get_yaxis().set_visible(False)
plt.imshow(np.random.random((10,10)))

buffer_ = StringIO()
plt.savefig( buffer_, format = "png", bbox_inches = 'tight', pad_inches = 0 )
buffer_.seek(0)

image = PIL.Image.open( buffer_ )

ar = np.asarray(image)
cv.imshow( 'a', ar )
cv.waitKey(0)
cv.destroyAllWindows()

在这里，程序终止后，我从 MS VC++ 运行时库中收到运行时错误。我更好的猜测是，这是因为开放的“buffer_”。但我得到了所需的输出。

---

编辑2：使用关闭缓冲区

buffer_.close()

没有解决运行时错误

---

最后编辑：解决方案

Answer 1

除非你真的需要标记刻度等，

im._rgba_cache

让您可以直接访问numpy作为颜色映射数据的 MxNx4 数组。

如果您只想要颜色映射数据，您可以imshow完全直接传递自己的颜色映射数据（请参阅选择颜色映射的指南）

my_cm = maplotlib.cm.get_cmap('Reds')
normed_data = (data - np.min(data)) / (np.max(data) - np.min(data))
mapped_data = my_cm(normed_data)

这将为您返回一个映射在0and之间的 MxNx4 数组1，

mapped_datau8 = (255 * my_cm(normed_data)).astype('uint8')

或者

mapped_data = my_cm(normed_data, bytes=True)

将其转换为无符号整数。

matplotlib包括一系列规范化代码，请参见此处。

get_cmap文档和颜色图库

编辑：在https://stackoverflow.com/a/14880947/380231指出的固定监督

Answer 2

您确定要转换方法的返回值还是整个情节？

对于后者，您应该尝试：

• 使用将绘图保存到 StringIO 缓冲区图像savefig
• 使用 PIL 或 opencv 从此缓冲区加载图像
• 将其转换为 numpy 数组

请参阅下面的示例：

import numpy as np
import matplotlib.pyplot as plt
import PIL
from cStringIO import StringIO

plt.imshow(np.random.random((20,20)))
buffer_ = StringIO()
plt.savefig(buffer_, format = "png")
buffer_.seek(0)
image = PIL.Image.open(buffer_)
ar = np.asarray(image)
buffer_.close()

查看savefig-*args以及**kwargs更多选项，例如 dpi、背景颜色、透明度、填充等。

如果您只想要彩色编码图像，没有轴、标签等，我仍然会这样做，只需使用

plt.subplots_adjust(0,0,1,1)

将轴延伸到整个图形。照顾好aspect你的情节，否则 mpl 可能会再次缩小你的轴。

Answer 3

受@Thorsten Kranz启发的更新答案

import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
import io
plt.imshow(np.random.random((20,20)))
with io.BytesIO() as buffer:
    plt.savefig(buffer, format = "png")
    buffer.seek(0)
    image = Image.open(buffer)
    ar = np.asarray(image)
print(ar.shape)

Answer 4

在这里，我找到了更多将数组转换为 rgba 的示例，我想使用岩浆 cmap 将频谱图转换为 rgb 图像，这就是我所做的；

import matplotlib.cm as cm
sm = cm.ScalarMappable(cmap='magma')
sm.set_clim(spect.min(), spect.max())
im = sm.to_rgba(spect)

plt.imshow(im)
plt.title(im.shape)