pytorch可视化工具visdom

visdom的github repo: https://github.com/facebookresearch/visdom

知乎一个教程：https://zhuanlan.zhihu.com/p/34692106

1、screen或者tmux，先开启visdom server

我比较习惯用screen，所以可以screen -S visdom_server，创建一个screen session，然后输入visdom

在本地浏览器输入ip地址加8097端口号，如128.21.32.54:8097，即可显示visdom窗口。

如果经常使用的是同一台服务器，由于端口号默认是8097，所以还可以将该地址添加书签，每一次使用时便可很方便地访问。

2、在python中的使用

上述开启了visdom的server之后，在python中先创建一个visdom对象

import visdom

vis = visdom.Visdom()

然后便可以方便地使用了。

一些注意点：

① visdom.image()显示的图像数组的格式是 通道数 x 高 x 宽，而像PIL.Image读取的图像是高 x 宽 x 通道数，因此需要对其numpy数组进行转置一下。

from PIL import Image
import numpy as np
import visdom

vis = visdom.Visdom()

img = Image.open('xxx.jpg')
img = np.array(img).transpose([2, 0, 1])

vis.image(img)

官网也提到了：

② 使用matplotlib需要设置backend模式，即matplotlib.use('agg')，否则直接在终端窗口import matplotlib.pyplot会报错

import matplotlib
matplotlib.use('agg')
import matplotlib.pyplot as plt
import visdom


plt.imshow(xxx)
visdom.matplot(plt)