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')