可视化YOLOv5的训练过程

YOLOv5集成了Weights & Biases Logging 🌟 NEW

Weights & Biases (W&B)是一个用于实时训练过程可视化和训练数据云端记录的工具，方便做训练数据的比较和研究，也方便团队协作。

详细使用步骤参考： YOLOv5 Weights & Biases Tutorial.

安装pip install wandb, 并在wandb官网上注册好账户后，先登录wandb

wandb login

然后执行YOLOv5训练程序

python train.py --img 640 --batch-size 32 --epochs 3 --data coco.yaml --weights yolov5s.pt --optimizer Adam --workers 6 --name yolov5s_20220404

启动YOLOv5训练 然后在wandb.ai上可以看到实时训练信息 实时训练信息 每完成一轮(Epoch)训练，数据就会更新一次，结果如下 wandb上的可视化训练过程