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