一、安装tensorboard
tensorboard原属于tensorflow模块组件,后pytorch接入了tensorboard的api,只需在pytorch的环境中执行以下命令。
pip install tensorboard
安装的过程可能会有些缓慢,耐心等待或者更换国内镜像源进行下载。
二、运行tensorboard
命令行输入
tensorboard --logdir=相对路径
这个相对路径指的是你训练生成的文件的路径。这里以yolov8为例,相对路径就是runs/detect/train6。
须知,使用tensorboard查看实时训练效果需要文件下包含下面这种文件。一般,安装tensorboard之后,再训练模型时会自动产生。
三、实时查看训练情况
命令行输入命令运行之后,会返回一个网址,点击打开网址,会出现下面图片的网站。
如果保持默认,网站不会自动刷新,需要修改。点击右上角的设置,也就是齿轮按钮,
勾选Reload data。关闭之后,网站就会自动刷新了。