coco2017与yolov5

记录自己下载coco数据集并跑通的过程。

下载coco2017数据集:
COCO数据集百度网盘
注:上面的数据集只包含图片,不包含标签
下载标签,
COCO数据集txt格式标签下载
注:上面的标签虽然annoations里是json格式,但是label文件下的标签已经转化为txt格式了,所以直接拿来用就行。

数据下载完成后,将数据集整合一下,参照下面的格式
coco2017与yolov5
最后:在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
上一篇:YOLOV5 服务器pytorch环境配置踩坑问题


下一篇:YOLOv5算法的部分笔记