我有一个形状为numpy的数组，(3, 256, 256)它是分辨率为256x256的3通道（RGB）图像。我想这与保存到磁盘Image从PIL通过执行以下操作：
from PIL import Image
import numpy as np
#... get array s.t. arr.shape = (3,256, 256)
img = Image.fromarray(arr, 'RGB')
img.save('out.png')
但是，这会将尺寸为256x3的图像保存到磁盘
@Dietrich答案是有效的，但是在某些情况下，它将翻转图像。由于转置运算符会反转索引，因此如果图像存储在RGB x rows x cols转置运算符中，将会产生cols x rows x RGB（这是旋转的图像，而不是所需的结果）。
>>> arr = np.random.uniform(size=(3,256,257))*255
注意257用于可视化目的。
>>> arr.T.shape
(257, 256, 3)
>>> arr.transpose(1, 2, 0).shape
(256, 257, 3)
最后一个是在某些情况下可能需要的，因为它会重新排列图像（rows x cols x RGB在示例中）而不是完全转置图像。
>>> arr = np.random.uniform(size=(3,256,256))*255
>>> arr = np.ascontiguousarray(arr.transpose(1,2,0))
>>> img = Image.fromarray(arr, 'RGB')
>>> img.save('out.png')
可能甚至不需要强制转换为连续数组，但是最好在保存图像之前确保图像是连续的。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句