python matplotlib 播放图片变化过程

最新推荐文章于 2025-01-13 05:55:35 发布

转载最新推荐文章于 2025-01-13 05:55:35 发布 · 356 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/subic/p/7929630.html

文章标签：

#python

本文介绍如何使用Python及matplotlib库将原始图片与经自编码网络处理后的图片进行对比展示的方法，并通过生成动图直观地展现图片变化过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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的代码。最后可能你会遇到这样的问题，