多通道 numpy 数组到 PIL 图像

数据挖掘 Python 计算机视觉 火炬 图片
2022-02-28 09:06:15

我有一个 4 通道 Numpy 图像,需要转换为 PIL 图像才能在图像上实现 torchvision 转换。但是当我尝试这样做时,PIL.Image.from_array(<my_numpy_image>)我收到以下错误。

TypeError: Cannot handle this data type

我现在不能松动和摆脱任何频道信息,因此无法丢弃任何频道。

1个回答

尝试指定模式,以便 PIL 知道数据格式。

img = Image.fromarray(source_array, mode="CMYK")

如果这不起作用,源数组的形状是什么?