YOLOv5模型训练训练可视化

YOLOv5模型训练训练可视化

安装相关包

pip install wandb

安装完成后开始训练，执行训练代码

!python train.py --data "/content/dataset/spore_0715/data.yaml" --cfg models/yolov5s.yaml --weights '' --batch-size 8 --rect --epoch 600 

已经创建过账号的输入2
点击链接访问站点 https://wandb.ai/authorize，复制后贴到终端中

点击view project 所显示的链接即可查看训练可视化