总结一下yolov5训练模型,目标检测常用的指令(更新,不完全正确,有错请指正)

1、activate torch107
激活环境,最好每个软件、每个cmd都激活一下环境。
2、python train.py --img 640 --batch 4 --epoch 300 --data ./data/A.yaml --cfg ./models/yolov5m.yaml --weights weights/yolov5m.pt --workers 0
训练模型指令,epoch300为迭代次数。
3、python detect.py --source data/images/shiyan.jpg --weights ./weights/last.pt --device 0 --save-txt
根据训练的模型,对图片进行检测,savetxt为获取目标框坐标信息。

版权声明:本文为CSDN博主「吕某人☔️」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_44533495/article/details/118339151

吕某人☔️

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐