1.确保imageio已经安装
pip install imageio
2.函数准备
def create_gif(image_list, gif_name): import imageio frames = [] for image_name in image_list: frames.append(imageio.imread(image_name)) # Save them as frames into a gif imageio.mimsave(gif_name, frames, 'GIF', duration=0.1)
3.函数调用
image_list = [] for step in range(20): i = i+1 sess.run(train) print('W=', sess.run(W), 'b=', sess.run(b), 'loss=', sess.run(loss)) plt.xlim((-2, 2)) plt.ylim((0.1, 0.5)) plt.scatter(x_data, y_data, c='r') plt.plot(x_data, sess.run(W)*x_data + sess.run(b)) plt.savefig("./static/"+str(i)+".png") plt.close() image_list.append("./static/"+str(i)+".png") create_gif(image_list, 'regression_res.gif')