目标检测 YOLOv5中使用COCO数据集

目标检测 YOLOv5中使用COCO数据集

flyfish

YOLOv5数据集转换工具下载地址

第一步
打开general_json2yolo.py

 if __name__ == '__main__':
    source = 'coco'

    if source == 'coco':
        convert_coco_json('../../Downloads/coco/annotations/')

改成自己的COCO文件所在目录

第二步
执行 python general_json2yolo.py

生成结果如下

├── new_dir
│   ├── images
│   └── labels
│       └── train2017

labels中的train2017存储是标注文本，例如000000000009.txt

第三步
可以把COCO的训练图像文件放到

├── new_dir
    ├── images
    │   └── train2017

和COCO128的目录结构一模一样