Pytorch可视化工具Visdom超简单入门

visdom的github repo: https://github.com/facebookresearch/visdom
知乎一个教程:https://zhuanlan.zhihu.com/p/34692106

上次写过Visdom在服务器端启动的博文:
https://blog.csdn.net/qq_18649781/article/details/89188326
这次简单介绍下在Python代码里用Visdom:(非常简单)

1.加载模块

import visdom

2.创建一个visdom环境

 vis = visdom.Visdom(env="visual eye", port=10041) # env后是环境名称,port是服务器可用的端口

3.显示图片

 img_path = "/xxxxx/xxxxx.jpg"
 img = cv2.imread(img_path)
 vis.image(img)

运行代码前,在服务器端开启visdom:

python -m visdom.server -p 10041

然后在浏览器输入:GPU服务器IP:端口号,即可启动
eg:XX.XXX.XX.XXX:10041
然后运行代码,效果图:Pytorch可视化工具Visdom超简单入门_第1张图片

你可能感兴趣的:(PyTorch框架学习,Python)