的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)