Pytorch使用visdom可视化问题

目录

• visdom安装
• 访问给出的地址
• 例子
• 完整代码
• 总结

visdom安装

pip install visdom

安装成功后，适应命令开启visdom服务，该服务基于Web，默认使用8097端口

python -m visdom.server

访问给出的地址

例子

完整代码

import torch
import visdom
 
 
#创建visdom客户端，使用默认端口8097，环境为first，环境的作用是对可视化的空间进行分区
vis=visdom.Visdom(env='first')
 
 
#vis对象有text（）、line（）和image（）等函数，其中的win参数代表了显示的窗格（pane）的名字
vis.text('Hello visdom',win='text1')
 
 
#在此处使用append为true来增加text，否则会覆盖之前的text
vis.text('hello Pytorch',win='text1',append=True)
 
#绘制y=-i^2+20Xi+1的曲线，opts可以进行标题、坐标轴标签等的配置
for i in range(20):
         vis.line(X=torch.FloatTensor([i]),Y=torch.FloatTensor([-i**2+20*i+1]),
             opts={'title':'y=-x^2+20x+l'},win='loss',update='append')
 
 
#可视化一张随机图片
vis.image(torch.randn(3,256,256),win='random_image')

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持小闻网。