Pytorch使用visdom进行可视化
文章目录
visdom介绍
visdom是Facebook专门为Pytorch开发的一款可视化工具,开源于2017年3月,项目地址为visdom,其十分轻量级,但功能丰富,提供了大多数的科学运算可视化API,基本可视化界面如下图所示:
visdom支持多种数据格式的可视化,包括数值、图像、文本以及视频等,支持Pytorch、Torch和Numpy。用户可以通过编程的方式组织可视化空间或者通过用户接口为数据打造仪表板,检查实验结果和调试代码。
visdom核心概念
-
env:环境
可以使用
envs
对可视化空间进行分区。默认地,每个用户都会有一个叫做main
的envs
。可以通过编程或UI
创建新的envs
。envs
的状态是长期保存的。不同环境的可视化结果相互隔离,互不影响。在使用时如果不指定env,默认使用main。不同用户、不同程序,最好使用不同的env,避免相互影响。 -
pane:窗格
窗格用于可视化图像、数值或文本等,它可以自由拖动、缩放、保存和关闭。类似于一个一个的小的窗口。一个程序可以使用同一个env中的不同pane,每个pane可视化记录某一信息。
visdom安装与使用
visdom安装十分简单,可以直接pip进行安装
pip install visdom
安装好以后,需要启动visdom服务
python -m visdom.server
或者(后台运行)
nohup python -m visdom.server &
visdom服务是一个Web Server服务,默认绑定8097端口。
基本使用,编程代码如下:
import visdom
import torch
# 新建一个连接客户端
# 指定env = 'test1'
vis = visdom.Visdom(env='test1')
x = torch.arange(1, 100, 0.01)
y = torch.sin(x)
vis.line(X=x,Y=y, win