YOLO V5

本文主要是介绍YOLO V5，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

1. 训练

python train.py --img 640 --batch 32 --epoch 1000 --data data/train_data.yaml --cfg models/yolov5m.yaml --weights weights/yolov5m.pt --device '0'

2. 检测

python detect.py --weights /root/yolov5/exp10_best.pt --source /root/train_data/images/test

3. 验证

python val.py  --data data/train_data.yaml --weights ./exp10_best.pt --augment

要注意修改train_data.yaml中的val项为test，才能检测测试集的AP，Recall

4. 参考

(24条消息) linux服务器虚拟环境部署yolov5 & 训练自己的数据集_Matilda55555的博客-CSDN博客_linux训练yolov5

(24条消息) yolov5训练自己的数据集并计算mAP_tpz789的博客-CSDN博客_yolov5计算map

(24条消息) YOLOv5训练自己的数据集（超详细完整版）_深度学习菜鸟的博客-CSDN博客_yolov5训练自己的数据集

这篇关于YOLO V5的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！