博客

tensorboard可视化可以实时跟进训练情况

话不多说，直接记录！！！

yolov5的可视化第一步找到yolo.py文件，将里面关于tensorboard的部分注释掉，如下图



第二步在终端输入tensorboard –logdir=./runs，因为yolov5在训练过程会自动生成run这个文件夹，所以在终端直接输入上面的命令就行了（前提是当前终端的文件目录位置要和train.py在同一目录下），当然了run文件里面记录的是所有训练的测试的结果，如果你要可视化当前训练的结果，你可以进入到当前的文件下面进行可视化，比如你训练的是第20个版本，那么tensorboard –logdir=./runs/train/exp20 就可以查看当前训练的可视化结果了

tensorboard报错

可能有的同学的tensorboard运行时报以下图中位置2的错误



出现这种错误的原因是重复插件即是安装了多个版本的tensorboard。

解决方法：

找到上图1中的位置fxs是我的虚拟环境，你只需要找到你的虚拟环境即可，虚拟环境下的lib/python3.8/site-packages/中把

tensorboard-2.8.0.dist-info 删除即可，你的tensorboard也可能是其他版本，总之你把带有.dist-info的那个tensorboard删除了即可

因为在安装tensorboard过程中，带有.dist-info后缀的版本也可能同时存在于tensorboard同级目录，将其删除即可。