记录自己下载coco数据集并跑通的过程。
下载coco2017数据集:
COCO数据集百度网盘
注:上面的数据集只包含图片,不包含标签
下载标签,
COCO数据集txt格式标签下载
注:上面的标签虽然annoations里是json格式,但是label文件下的标签已经转化为txt格式了,所以直接拿来用就行。
数据下载完成后,将数据集整合一下,参照下面的格式
最后:在spyder下:
run train.py --data data/data.yaml --cfg yolov5m.yaml --weights weights/yolov5m.pt --epochs 50 --batch-size 32
若是pycharm,
python train.py --data data/data.yaml --cfg yolov5m.yaml --weights weights/yolov5m.pt --epochs 50 --batch-size 32