目标检测 YOLOv5中使用COCO数据集
flyfish
第一步
打开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的目录结构一模一样