如何用yolov5训练自己的图片

首先采用脚本工具转换好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内存不足，好吧有小伙伴知道怎么解决吗