yolo命令

就叫飞六吧

于 2024-07-21 16:45:28 发布

阅读量121

点赞数 2

文章标签： YOLO 深度学习机器学习

本文链接：https://blog.csdn.net/weixin_52236586/article/details/140590209

版权

训练命令：
python .\train.py --data .\data\coco_mask.yaml --cfg .\models\yolov5s_mask.yaml --weights .\weights\yolov5s.pt --batch-size 20 --epochs 1  --name yolo_test-gpu --project yolo_test

python .\train.py --data .\data\coco_mask.yaml --cfg .\models\yolov5s_mask.yaml --weights .\weights\yolov5s.pt --batch-size 20 --epochs 20  --name yolo_test-gpu --project yolo_test

验证命令(图片、视频)：
python detect.py --source ./data/videos/1.mp4 --weights weights/yolov5s.pt --conf-thres 0.4

python detect.py --source ./data/images/ --weights weights/yolov5s.pt --conf-thres 0.4



python detect.py --source ./data/images/ --weights yolo_test/yolo_test-gpu3/weights/best.pt --conf-thres 0.3
python detect.py --source ./data/videos/1.mp4 --weights yolo_test/yolo_test-gpu3/weights/best.pt --conf-thres 0.3