简介

将YOLOv5格式数据集转换为COCO格式。

支持标注格式为矩形框和多边形框。

YOLOv5数据结构如下

  yolov5_dataset
├── classes.txt
├── non_labels  # 通常用来放负样本
│   └── bg1.jpeg
├── images
│   ├── images(13).jpg
│   └── images(3).jpg
├── labels
│   ├── images(13).txt
│   └── images(3).txt
├── train.txt
└── val.txt
  

转换

  yolov5_to_coco --data_dir dataset/yolov5_dataset --mode_list train,val
  
  • --data_dir: 数据集所在目录。示例为dataset/yolov5_dataset
  • --save_dir: 保存转换后的数据集目录。默认为dataset/yolov5_dataset_coco
  • --mode_list: 指定划分的数据集种类。 (例如:train,val,test / train,val)

转换后结构如下:

  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 . history