如果在无桌面的命令行环境使用matplotlib.pyplot绘图会遇到如下的问题:
RuntimeError: Invalid DISPLAY variable
此时,可以通过添加以下命令避免:
import matplotlib.pyplot as plt
plt.switch_backend('agg')
然后利用plt.imshow绘图:
plt.figure()
plt.title('FitureName')
plt.imshow(img)
最后通过savefig函数存储图片以便查看:
plt.savefig('输出路径')