有没有一种方法可以将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图像并对其进行进一步处理。我正在使用python
2.7，32位。

请帮助

谢谢

编辑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)

这将为您提供映射到0和之间的MxNx4数组1，

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

要么

mapped_data = my_cm(normed_data, bytes=True)

会将其转换为无符号整数。

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

get_cmapdoc和colormap画廊