首先配置yolov5环境,详细看这篇https://blog.csdn.net/qq_16792139/article/details/
首先采用脚本工具转换好xml为txt,脚本工具上传至群文件了 Q群477592543, yolo5totxt.py
配置官方文档查看github源地址https://github.com/ultralytics/yolov5/wiki/Train-Custom-Dat
数据集格式
记得修改yaml里面的类名和类数目,还有train.py里面的80替换成你的类数
然后运行
python train.py --img 640 --batch 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ''
就可以了。
然而,报错
cuda内存不足,好吧有小伙伴知道怎么解决吗