服务器之家

服务器之家 > 正文

Pytorch可视化之Visdom使用实例

时间:2021-12-17 12:46     来源/作者:ting_qifengl

一、visdom简介

visdom是由facebook公司开发的一个进行数据可视化的web应用程序,支持torch、numpy、pytorch这3个库的创建、管理和分享实时的数据可视化结果。

二、安装和运行

可直接使用pip进行安装,命令如下:

?
1
pip install visdom

安装过程如下:

Pytorch可视化之Visdom使用实例

安装完成后,运行下面的代码启动visdom服务器(运行需要时间,需要稍等一下;下图中的报错error我没有管,不影响正常运行):

?
1
python -m visdom.server

Pytorch可视化之Visdom使用实例

按照提示,在浏览器中输入http://localhost:8097就可以访问visdom了。初始界面如下图所示,是一个没有任何视窗的main环境。

Pytorch可视化之Visdom使用实例

三、可视化例子

1、输出hello world!

打开开发工具(我用的是pycharm2020.3),新建程序文件,输入以下代码:

?
1
2
3
4
import visdom
 
vis = visdom.visdom()
vis.text("hello world!")

点击运行,返回查看浏览器。这时浏览器会出现一个视窗,内容为“hello world!”。

Pytorch可视化之Visdom使用实例

 拖拽视窗右下角箭头处(如图绿框处)可对视窗大小进行缩放,拖动视窗顶部横条(如图红框处)可以移动视窗。

Pytorch可视化之Visdom使用实例

2、显示图像

visdom可以直接显示tensor格式的图像数据。

打开开发工具,新建程序文件,输入以下代码:

?
1
2
3
4
5
6
7
8
import visdom
from pil import image
import torchvision.transforms.functional as tf
 
demo_pic = image.open("1.jpg")
img_tensor = tf.to_tensor(demo_pic)
vis = visdom.visdom()
vis.image(img_tensor)

结果如下图所示:

Pytorch可视化之Visdom使用实例

 可以用images()函数同时显示多张图片,代码如下:

?
1
2
3
4
5
6
7
8
9
10
import visdom
from pil import image
import torchvision.transforms.functional as tf
import torch
 
demo_pic = image.open("1.jpg")
img_tensor = tf.to_tensor(demo_pic)
img_tensors = torch.tensor([img_tensor.numpy(), img_tensor.numpy()])
vis = visdom.visdom()
vis.images(img_tensors)

结果如下:

Pytorch可视化之Visdom使用实例

3、绘制散点图

使用scatter()函数可以绘制二维或者三维的散点图。代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import visdom
import numpy as np
import torch
 
y = np.random.rand(100)
vis = visdom.visdom()
# vis = visdom.visdom(env='my_wind')#设置环境窗口的名称是'my_wind',如果不设置默认为main
# 2—d
twod_scatter = vis.scatter(x=torch.rand(100, 2),
                          y=(y + 1.5).astype(int), # 转换成100个1或2的整数
                          opts=dict(
                              legend=['apples', 'pears'], # 图例名称
                              xtickmin=-1, # 设置x坐标轴下限
                              xtickmax=2.5, # 设置x坐标轴上限
                              xtickstep=0.5, # 设置x坐标轴间隔
                              ytickmin=-1,
                              ytickmax=2.5,
                              ytickstep=0.5,
                              markersymbol='dot' # 设置数据显示样式
                          ),
                          )
# 3-d
# 3d scatterplot with custom labels and ranges
threed_scatter = vis.scatter(x=np.random.rand(100, 3),
                             y=(y + 1.5).astype(int),
                             opts=dict(
                                 legend=['men', 'women'],
                                 markersize=5, # 标记大小
                                 xtickmin=0,
                                 xtickmax=2,
                                 xlabel='arbitrary', # 标签
                                 xtickvals=[0, 0.75, 1.6, 2], # 设置坐标轴显示值
                                 ytickmin=0,
                                 ytickmax=2,
                                 ytickstep=0.5,
                                 ztickmin=0,
                                 ztickmax=1,
                                 ztickstep=0.5,
                             )
                             )

运行结果如下图所示:

Pytorch可视化之Visdom使用实例

  如果想要通过程序实现散点图参数的更新,可以使用update_window_opts()函数,代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
vis.update_window_opts(
    win=twod_scatter,
    opts=dict(
        legend=['apples', 'pears'],
        xtickmin=0,
        xtickmax=1,
        xtickstep=0.5,
        ytickmin=0,
        ytickmax=1,
        ytickstep=0.5,
        markersymbol='dot'
    )
)

update_window_opts()传入两个参数,第一个是视窗的实例,例子中是twod_scatter;第二个是更新的参数字典opts。

更新后结果如下图所示:

Pytorch可视化之Visdom使用实例

4、绘制线条

4.1 绘制一条直线

代码如下:

?
1
2
3
4
5
6
7
8
import visdom
import numpy as np
vis = visdom.visdom(env='my_windows'# 设置环境窗口的名称,如果不设置名称就默认为main
# 绘制一条直线
x = list(range(10))
y = list(range(10))
# 使用line函数绘制直线 并选择显示坐标轴
vis.line(x=np.array(x), y=np.array(y), opts=dict(showlegend=true))

结果如图所示:

Pytorch可视化之Visdom使用实例

4.2 绘制两条直线

代码如下:

?
1
2
3
4
5
6
7
8
# 绘制两条直线
import visdom
import numpy as np
vis = visdom.visdom(env='my_windows')
x = list(range(10))
y = list(range(10))
z = list(range(1,11))
vis.line(x=np.array(x), y=np.column_stack((np.array(y), np.array(z))),  opts=dict(showlegend=true))

结果如图所示:

Pytorch可视化之Visdom使用实例

4.3 绘制正弦曲线

代码如下:

?
1
2
3
4
5
6
import visdom
import torch
vis = visdom.visdom(env='sin')
x = torch.arange(0, 100, 0.1)
y = torch.sin(x)
vis.line(x=x,y=y,win='sin(x)',opts=dict(showlegend=true))

结果如图所示:

Pytorch可视化之Visdom使用实例

参考文献

pytorch深度学习入门--曾芃壹

https://www.pythonf.cn/read/3068

总结

到此这篇关于pytorch可视化之visdom使用的文章就介绍到这了,更多相关pytorch可视化visdom内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/ting_qifengl/article/details/119443813

标签:

相关文章

热门资讯

yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全
2020微信伤感网名听哭了 让对方看到心疼的伤感网名大全 2019-12-26
2021年耽改剧名单 2021要播出的59部耽改剧列表
2021年耽改剧名单 2021要播出的59部耽改剧列表 2021-03-05
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总
苹果12mini价格表官网报价 iPhone12mini全版本价格汇总 2020-11-13
返回顶部