yolov5使用tensorboard可视化训练结果

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

话不多说,直接记录!!!
yolov5的可视化第一步找到yolo.py文件,将里面关于tensorboard的部分注释掉,如下图
yolov5使用tensorboard可视化训练结果_第1张图片
第二步在终端输入tensorboard --logdir=./runs,因为yolov5在训练过程会自动生成run这个文件夹,所以在终端直接输入上面的命令就行了(前提是当前终端的文件目录位置要和train.py在同一目录下),当然了run文件里面记录的是所有训练的测试的结果,如果你要可视化当前训练的结果,你可以进入到当前的文件下面进行可视化,比如你训练的是第20个版本,那么tensorboard --logdir=./runs/train/exp20 就可以查看当前训练的可视化结果了

tensorboard报错

可能有的同学的tensorboard运行时报以下图中位置2的错误
yolov5使用tensorboard可视化训练结果_第2张图片
出现这种错误的原因是重复插件即是安装了多个版本的tensorboard。
解决方法:
找到上图1中的位置fxs是我的虚拟环境,你只需要找到你的虚拟环境即可,虚拟环境下的lib/python3.8/site-packages/中把
tensorboard-2.8.0.dist-info 删除即可,你的tensorboard也可能是其他版本,总之你把带有.dist-info的那个tensorboard删除了即可
因为在安装tensorboard过程中,带有.dist-info后缀的版本也可能同时存在于tensorboard同级目录,将其删除即可。

你可能感兴趣的:(深度学习,pytorch,python)