YOLO V5

0.安装依赖项

pip install -r requirements.txt

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 --data data/train_data.yaml --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训练自己的数据集