如何将3通道numpy数组另存为图像

如何解决如何将3通道numpy数组另存为图像

@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')

可能甚至不需要强制转换为连续数组，但是最好在保存图像之前确保图像是连续的。

解决方法

我有一个形状为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)
img = Image.fromarray(arr,'RGB')
img.save('out.png')

但是，这会将尺寸为256x3的图像保存到磁盘