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)