在工作中,大部分都是使用VScode编写代码,并通过SSH远程连接服务器,时刻将代码和数据放在服务器上。
在进行图像处理和跑深度学习模型过程中,想在本地查看服务器上的图片处理效果(im.show())或模型指标变化结果图片(plt.scatter()等),是没办法使用这些指令(im.show()、plt.scatter()),在本地桌面看图片的。
不过有个方法,可以把图片保存下来,再使用VScode打开查看
下面记录一下VSCode+SSH环境下显示matplotlib绘制的figure
1. 代码
plt.switch_backend('agg') 必须加上这行代码,不然不能将图片保存到centos
import numpy as np
import matplotlib.pyplot as plt
plt.switch_backend('agg')
from sklearn.datasets.samples_generator import make_blobs
X, y = make_blobs(n_samples=1000, n_features=2, centers=[[-1,-1], [0,0], [1,1], [2,2]], cluster_std=[0.4, 0.2, 0.2, 0.2],
random_state =9)
plt.scatter(X[:, 0], X[:, 1], marker='o')
plt.savefig("./1.jpg")
2 . 图片显示