On this page
YOLOv5 YAML → COCO
简介
将以yaml文件给出的YOLOv5格式数据集转换为COCO格式
支持标注格式为矩形框和多边形框。
YOLOv5 yaml结构如下:
具体结构示例文件,可移步:yolov5_yaml_dataset
yolov5_yaml_dataset
├── images
│ ├── train
│ │ ├── images(13).jpg
│ │ └── images(3).jpg
│ └── val
│ ├── images(13).jpg
│ └── images(3).jpg
├── labels
│ ├── train
│ │ ├── images(13).txt
│ │ └── images(3).txt
│ └── val
│ ├── images(13).txt
│ └── images(3).txt
└── sample.yaml
转换
yolov5_yaml_to_coco --yaml_path dataset/yolov5_yaml_dataset/sample.yaml
--yaml_path
: yaml文件路径--save_dir
: 保存转换后的数据集目录。默认为dataset/yolov5_yaml_dataset_coco
转换后结构如下:
具体结构示例文件,可移步:COCO_dataset
COCO_dataset
├── annotations
│ ├── instances_train2017.json
│ └── instances_val2017.json
├── train2017
│ ├── 000000000001.jpg
│ └── 000000000002.jpg
└── val2017
└── 000000000001.jpg
Last updated 24 Apr 2024, 13:58 +0800 .