如何将numpy.ndarray数据转换为所需的图像格式？

Question

我有一个numpy.ndarray格式的数据，例如[45 42 35 ... 53 29 31]（uint8s的1x3072 numpy数组），该数组的行存储32x32彩色图像。前1024个条目包含红色通道值，接下来的1024个包含绿色，最后的1024个包含蓝色。

我想将该ndarray图像转换为jpg格式的32x32彩色图像。 尝试了许多选项，但似乎没有创建有效/正确的图像。

是否有任何建议将图像转换为32x32形状？

from PIL import Image
import numpy as np

img_data=[45 42 35 ... 53 29 31] 
data_2 = np.resize(img_data,(32,32,3))
img = Image.fromarray(data_2,'RGB')
img.save('./my.png')

Answer 1

这吗？

from PIL import Image
import numpy as np

img_data=[45 42 35 ... 53 29 31] 
data_2 = np.swapaxes(np.resize(img_data,(3,32,32)),0, 2)
img = Image.fromarray(data_2,'RGB')
img.save('./my.png')