- 在你的mmyolo环境下,安装tensorboard
pip install tensorboard
- 在你跑的实验配置文件中,加入关于tensorboard的配置。例如在
configs/mydataset/mymodel.py
里面加入
visualizer = dict(vis_backends=[dict(type='LocalVisBackend'),dict(type='TensorboardVisBackend')])
- 运行训练命令,把
mymodel.py
跑起来,Tensorboard 文件会生成在可视化文件夹work_dirs/mymodel/${TIMESTAMP}/vis_data
下,之后在命令行中输入:
tensorboard --logdir=work_dirs/mymodel
然后会输出下面这种:
TensorBoard 2.14.0 at http://localhost:6006/ (Press CTRL+C to quit)
- 按照如图所示配置隧道,这一步非常重要哦!
- 在本地访问
http://localhost:6006/
即可查看训练可视化情况,如图所示