mmdetection2.0中利用visdom绘制mAP动态图
程序员文章站
2023-08-11 19:51:59
Vidsom是Facebook专门为PyTorch开发的一款可视化工具 ,可以直接对Tensor进行操作。能够胜任大部分的数据可视化任务。关于visdom的详细介绍请参考PyTorch的远程可视化神器visdom在mmdetection中想要完成动态可视化过程,需要用到Hook。关于Hook,其实就是在不改变网络输入输出的结构的前提下,方便地获取、改变网络中间层变量的值和梯度。更详细的资料请参考半小时学会 PyTorch Hook下面是在mmdetection中利用visdom绘制mAP动态图的方法步骤...
Vidsom是Facebook专门为PyTorch开发的一款可视化工具 ,可以直接对Tensor进行操作。能够胜任大部分的数据可视化任务。关于visdom的详细介绍请参考PyTorch的远程可视化神器visdom
在mmdetection中想要完成动态可视化过程,需要用到Hook。关于Hook,其实就是在不改变网络输入输出的结构的前提下,方便地获取、改变网络中间层变量的值和梯度。更详细的资料请参考半小时学会 PyTorch Hook
下面是在mmdetection中利用visdom绘制mAP动态图的方法步骤。
- 首先创建visualization.py文件
import visdom
import numpy as np
from mmcv.runner import Hook
from torch.utils.data import DataLoader
class Visualizer(Hook):
def __init__(self, dataloader, interval=1, **eval_kwargs):
if not isinstance(dataloader, DataLoader):
raise TypeError('dataloader must be a pytorch DataLoader, but got'
f' {type(dataloader)}')
self.vis = visdom.Visdom(env='test')
self.flag={}
self.log_text = ""
def after_train_epoch(self, runner):
self.plot_many(runner.log_buffer.output, runner.epoch)
def plot_many(self, d,epoch):
for k, v in d.items():
if type(v) is float:
self.plot(k, epoch,v)
def plot(self, name, x, y, **kwargs):
flag = self.flag.get(name,True)
self.vis.line(Y=np.array([y]), X=np.array([x]),
win=name,
opts=dict(title=name),
update=None if flag else 'append',
**kwargs)
self.flag[name]=False
将文件放入一个你指定的文件夹中,并更改对应的init.py文件,例如我放到了mmdet/utils/下。
- 更改mmdet/apis/train.py文件
导入模块
from mmdet.utils import Visualizer
在验证集部分代码注册钩子Hook
vis_hook = Visualizer
runner.register_hook(vis_hook(val_dataloader, **eval_cfg))
- 训练结果展示
确保所在的虚拟环境中安装了visdom
$ pip install visdom
开启visdom服务
(open-mmlab) username@name:~$ python -m visdom.server
/home/dreamdeck/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/visdom/server.py:39: DeprecationWarning: zmq.eventloop.ioloop is deprecated in pyzmq 17. pyzmq now works with default tornado and asyncio eventloops.
ioloop.install() # Needs to happen before any tornado imports!
Checking for scripts.
It's Alive!
INFO:root:Application Started
You can navigate to http://localhost:8097
复制最后一行的地址,打开浏览器并打开地址。开启训练,并选择test环境
随着训练的进展,图也会逐渐变化。
本文地址:https://blog.csdn.net/weixin_38369492/article/details/107460595
上一篇: 6大步骤,快速掌握说明型软文写作方法
下一篇: 总结常见的网站降权原因