python：如何让matplotlib不显示图窗但保存图片

在一些服务器上，我们跑的模型会让输入的图片生产新的图片，并绘制出来。常见的方法有plt.imshow等。
不过，如果服务器没有GUI界面，到这一步可视化会出现报错。
解决办法是在import包的部分加上下述语句：

import matplotlib  
matplotlib.use('Agg') 

该plt的地方还plt就行，如果你调用了plt.imshow，还保存了图片到某个路径，那么运用上述返回发，你就能发现，运行完后不但没有报错，而且图片还保存了~