服务器远程visdom可视化

需求描述

ssh连接服务器，在服务器上训练深度学习模型，希望在本地可视化训练数据等

方法

1. 服务器端安装visdom

在对应python环境下安装visdom：

pip install visdom

2. 本地ssh连接服务器并映射端口

运行命令：

ssh -L <本地端口>:localhost:8097 -p <ssh访问服务器的端口> <服务器用户名>@<ssh访问服务器的ip>

e.g 若指定本地端口为8080(本地端口可任意指定)，服务器用户名为ubuntu，端口为40008，服务器ip为47.100.99.121，则运行：

ssh -L 8080:localhost:8097 -p 40008 ubuntu@47.100.99.121

3. 服务器端启动visdom

在完成第二步后，在服务器端启动visdom服务：

python -m visdom.server

你将看到：

It's Alive!
INFO:root：Application Started
You can navigate to http://localhost:8097

4. 本地端访问visdom窗口

浏览器访问：localhost:<设定的本地的端口>
在上面的例子中，应访问：localhost:8080

页面顶端Environment后的为visdom窗口名称，通常默认为main

5.服务器端visdom操作

以读取显示图片为例
在服务器端上新开一个窗口，运行visdom：

from visdom import Visdom
import cv2

img=cv2.imread('img_path')
vis=Visdom(env='main') # main为对应的visdom窗口名称
vis=image(img.transpose(2,0,1)[::-1,...])

6.你将本地端visdom窗口看到显示的图片

完成