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,复制后贴到终端中
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/aa2ed4560c3a859c05acfdf74eaf5291.png)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/d1592451f0d8ae8ee7e663a9f8427ada.png)
点击view project 所显示的链接即可查看训练可视化
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/95e366db75b507c67f9258f1207dbb27.png)