python matplotlib 播放图片变化过程

Posted 2020-10-16 subic

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

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(‘filename.npy‘)
fig = plt.figure()
plt.subplot(1, 2, 1)
plt.imshow(img_ori)
plt.subplot(1, 2, 2)
ims = []
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‘)
plt.show()

运行结果是