其他分享
首页 > 其他分享> > coco2017与yolov5

coco2017与yolov5

作者:互联网

记录自己下载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

标签:yolov5,yolov5m,--,标签,yaml,weights,coco2017,data
来源: https://blog.csdn.net/qq_45988298/article/details/115212331