1

我正在使用 imageio 将 png 图像写入文件。

import numpy as np
import matplotlib.cm as cm
import imageio # for saving the image
import matplotlib as mpl

hm_colors = ['blue', 'white','red']
cmap = mpl.colors.LinearSegmentedColormap.from_list('bwr', hm_colors)
data = np.array([[1,2,3],[5,6,7]])
norm = mpl.colors.Normalize(vmin=-3, vmax=3)
colormap = cm.ScalarMappable(norm=norm, cmap=cmap)
im = colormap.to_rgba(data)
# scale the data to a width of w pixels
im = np.repeat(im, w, axis=1)
im = np.repeat(im, h, axis=0)
# save the picture
imageio.imwrite("my_img.png", im)

这个过程是自动执行的,我注意到一些错误消息说:

Error closing: 'Image' object has no attribute 'fp'.

在此消息之前,我收到警告:

/usr/local/lib/python2.7/dist-packages/imageio/core/util.py:78: UserWarning: Lossy conversion from float64 to uint8, range [0, 1] dtype_str, out_type.__name__))

但是,图像似乎生成并保存得很好。我找不到数据来重新创建此消息。

知道为什么我会收到此错误以及为什么它不会明显影响结果吗?我不使用 PIL。一个可能的原因可能来自在 Celery 中使用它。

谢谢!L.

4

1 回答 1

0

imageio.imwrite我在 Python 3.5 中使用时遇到了同样的问题。这是一个相当无害的,除了它停止垃圾收集并在写入数千张图像时导致过多的内存使用。解决方案是使用PIL模块,它是imageio. 代码的最后一行应为:

from PIL import Image
image = Image.fromarray(im)
image.save('my_img.png')
于 2018-03-21T09:58:17.810 回答