Python - Numpy RGB像素阵列到图像

Question

我有一个像素数组

np.shape(pred2)
Out[35]: (3000, 3, 32, 32)

它有3000个图像，3个值rgb，每个图像的大小为32 * 32。 我想从中创建一个图像。

这是我到目前为止：

img = Image.new( 'RGB', (32,32), "black") # create a new black image
pixels = img.putdata(pred2[1,:])

任何人都可以帮我解决一下我做错了什么吗？

Answer 1

图像是形状(h, w, 3) ，而不是(3, h, w) 。 您需要相应地置换轴。 根据您是否关心宽度与高度，看起来您可以这样做：

im = pred2[1].T
scipy.misc.imsave('the_image_file.png', im)