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')总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持代码网。
发表评论