我正在使用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。 一个可能的原因可能来自在芹菜中使用它。
谢谢! 升。
答案 0 :(得分:0)
我在Python 3.5中使用imageio.imwrite
遇到了同样的问题。这是一个相当无害的事情,因为它正在停止垃圾收集并在编写数千张图像时导致过多的内存使用。解决方案是使用PIL
模块,它是imageio
的依赖项。代码的最后一行应为:
from PIL import Image
image = Image.fromarray(im)
image.save('my_img.png')