简介

一键转换labelme标注的数据格式为COCO格式

labelme结构如下:

  labelme_dataset
├── val_0001.jpg
├── val_0001.json
├── val_0002.jpg
└── val_0002.json
  

转换

  labelme_to_coco --data_dir dataset/labelme_dataset \
                --save_dir dataset/coco_dataset \
                --val_ratio 0.2 \
                --have_test \
                --test_ratio 0.2
  
  • --data_dir: 数据集所在目录。示例为dataset/labelme_dataset
  • --save_dir: 保存转换后的数据集目录。默认为dataset/labelme_dataset_coco
  • --val_ratio: 验证集所占比例,示例为总量的0.2
  • --have_test: 是否划出测试集,示例为False,如果想要划分测试集,直接加上该参数即可。
  • --test_ratio: 测试集的比例,示例为总量的0.2

转换后结构如下:

  COCO_dataset
├── annotations
│   ├── instances_train2017.json
│   └── instances_val2017.json
├── train2017
│   ├── 000000000001.jpg
│   └── 000000000002.jpg
└── val2017
    └── 000000000002.jpg
  

Last updated 24 Apr 2024, 13:58 +0800 . history