python – 如何将3通道numpy数组保存为图像

我有一个形状为numpy的阵列(3,256,256),这是一个3通道(RGB)图像的重新分离256×256.我试图通过执行以下操作将此文件从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')

但是,这会将尺寸为256×3的图像保存到磁盘

最佳答案
@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)

最后一个是您在某些情况下可能需要的,因为它重新排序图像(示例中的行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')

可能甚至不需要转换为连续阵列,但最好在保存之前确保图像是连续的.

转载注明原文:python – 如何将3通道numpy数组保存为图像 - 代码日志