Pytorch使用visdom可视化

  • Post author:
  • Post category:其他


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')





版权声明:本文为qq_41251963原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。