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
暂无评论