python matplotlib 播放图片变化过程

偏执的太偏执、 2021-11-02 02:50 373阅读 0赞

最近想将原图片和处理后的图片放在一起观察图片的变化过程。但是网上并么有找到有用的示例代码，所以粘出来和大家分享一下。

import numpy as np
    import matplotlib.pyplot as plt
    from matplotlib import animation
    from 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

运行结果是

![899363-20171130163624104-877581000.gif][]

转载于:https://www.cnblogs.com/subic/p/7929630.html

[899363-20171130163624104-877581000.gif]: /images/20211102/02de41c3fff94b3abcfedd922b58878b.png