Python - Numpy RGB 像素阵列到图像

Python - Numpy RGB pixel array to image

我有一个像素数组

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,:])

谁能帮我看看我做错了什么?

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

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