最近想将原图片和处理后的图片放在一起观察图片的变化过程。但是网上并么有找到有用的示例代码,所以粘出来和大家分享一下。
import numpy as npimport matplotlib.pyplot as pltfrom matplotlib import animationfrom PIL import Image
img_ori = np.array(Image.open('5.png')) img_tran = np.load('img_tran.npy') fig = plt.figure() plt.subplot(1, 2, 1) plt.imshow(img_ori) plt.subplot(1, 2, 2) i = [i for i in range(200) if i % 5 == 0] def update(i): return plt.imshow(img_tran[i].reshape([28, 28])) ani = animation.FuncAnimation(fig, update) # ani.save('contrast.mp4', fps=15, # extra_args=['-vcodec', 'libx264'], # writer='ffmpeg_file') ani.save('contrast.gif', writer='imagemagick', fps=30) # plt.show()
简单分析一下,输入是一张手写数字的图片,实际上我是将这张图片经过了一个自编码网络,通过求输入与输出的距离,然后反向传播迭代权重和偏置。最后形成一张接近原图的图像。具体的就不细讲了,这里我主要是讲如何显示出来这些数据。我将迭代生成的图像保存在‘img_train.npy’文件里。
i = [i for i in range(200) if i % 5 == 0]
其实,img_tran是一个维度为(1000,784, 1)的数组,我就显示了其中一部分,i 也就是一个索引列表,索引要显示的图像。后面就是利用animation.FuncAnimation函数显示动图,该函数的具体细节可以自行google。然后由于博客园上传不了视频,我就在下面加了一行保存为gif的代码。最后可能你会遇到这样的问题,
KeyError: 'ffmpeg'
安装这个库就行。具体参考http://www.jianshu.com/p/c41d4127cc7d
运行结果是