从列表到np数组转换为JPEG时出错

时间:2020-05-21 15:53:11

标签: python numpy python-imaging-library jpeg

在我的代码中,我需要将图像转换为numpy数组,然后从数组转换为列表。对列表进行一些更改后,我需要转换回图像,但出现此错误

    Traceback (most recent call last):
  File "/home/owner/anaconda3/envs/proj1/lib/python3.7/site-packages/PIL/Image.py", line 2714, in fromarray
    mode, rawmode = _fromarray_typemap[typekey]
KeyError: ((1, 1, 3), '<i8')

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/owner/PycharmProjects/arrays3/test.py", line 31, in <module>
    im = Image.fromarray(b)
  File "/home/owner/anaconda3/envs/proj1/lib/python3.7/site-packages/PIL/Image.py", line 2716, in fromarray
    raise TypeError("Cannot handle this data type: %s, %s" % typekey)
TypeError: Cannot handle this data type: (1, 1, 3), <i8

我知道该错误是由于从数组到列表以及返回的过渡而发生的,但是我不确定为什么。这里的一些代码会产生相同的错误,但由于print语句返回true,因此未对图像数据的内容进行任何修改。

im = Image.open("wp2793461-windows-98-wallpaper-pack.jpg")
a = np.asarray(im)
lst = a.tolist()
b = np.asarray(lst)
print(np.array_equal(a, b))
im = Image.fromarray(b)
im.save("new.jpg")

1 个答案:

答案 0 :(得分:1)

好难题!我查看了ab之间的区别,发现numpy的dtype两者都不同。

>>> print(a.dtype)
uint8
>>> print(b.dtype)
int64

如果您通过以下方式创建b,它将再次起作用:

b = np.asarray(lst, dtype=a.dtype)