python：如何讓matplotlib不顯示圖窗但保存圖片

在一些服務器上，我們跑的模型會讓輸入的圖片生產新的圖片，並繪製出來。常見的方法有plt.imshow等。
不過，如果服務器沒有GUI界面，到這一步可視化會出現報錯。
解決辦法是在import包的部分加上下述語句：

import matplotlib  
matplotlib.use('Agg') 

該plt的地方還plt就行，如果你調用了plt.imshow，還保存了圖片到某個路徑，那麼運用上述返回發，你就能發現，運行完後不但沒有報錯，而且圖片還保存了~