Pytorch的远程可视化工具Visdom安装及使用

Pytorch的远程可视化工具Visdom安装及使用

1.visdom安装

在本机python程序的终端中进行安装

pip install visdom

开启服务

python -m visdom.server

2.创建visdom新环境（按照写python代码的方式来就行）

vis = visdom.Visom(env='test_1')

创建完环境在浏览器中输入 http://localhost:8097查看当前环境

3.文本和图像测试

• 文本

vis.text('Hello World', win='text1')
vis.text('Hi', win='text1', append=True)#在Hello World后进行补充

• 作图
  （1）绘制正弦图像

import visdom
import torch

vis = visdom.Visdom(env='test_1')

# 绘制正弦函数
x = torch.arange(1, 100, 0.01)
y = torch.sin(x)
vis.line(X=x,Y=y, win='sinx',opts={'title':'y=sin(x)'})#设置x,y轴和图标题
# 绘制36张图片随机的彩色图片
vis.images(torch.randn(36,3,64,64).numpy(),nrow=6, win='imgs',opts={'title':'imgs'})

（2）绘制直线图像

import visdom
import torch

# 新建001环境
vis = visdom.Visdom(env='001')


#绘制直线
vis.line([0.],  # Y的第一个点
         [0.],  # X的第一个点
         win="line",  # 右上角窗口的名称
         opts=dict(title='line')  # opt的参数都可以用python字典的格式传入，还有很多其他的类似matplotlib美化图形的参数参考官网
         )

vis.line([1, ],  # Y的下一个点
         [1.],  # X的下一个点
         win="line",
         update='append'  # 添加到下一个点后面
         )