yolov5的tensorboard版本
时间: 2023-06-30 15:04:15 浏览: 120
yolov5tensorrt_v6.0.rar
5星 · 资源好评率100%
YOLOv5 提供了 TensorBoard 的支持,可以通过安装 TensorBoard 来进行可视化训练过程的监控。
在安装了 PyTorch 和 YOLOv5 后,在命令行输入以下命令即可安装 TensorBoard:
```
pip install tensorboard
```
安装完成后,在训练命令中加入 `--tensorboard` 参数即可启动 TensorBoard:
```
python train.py --img 640 --batch 16 --epochs 50 --data coco.yaml --weights yolov5s.pt --cache --name yolov5s_results --tensorboard
```
然后在命令行输入以下命令启动 TensorBoard:
```
tensorboard --logdir runs
```
此时,TensorBoard 就会监听本地的 6006 端口,可以在浏览器中输入 `http://localhost:6006/` 访问 TensorBoard 页面,查看训练过程的可视化信息。
阅读全文