YOLO V5

2022/4/18 6:14:42

本文主要是介绍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的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程