简介

将COCO格式数据集转换为可以直接用labelImg工具可视化标注的YOLO格式。

COCO结构如下： 

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

转换 

  coco_to_labelImg --data_dir dataset/COCO_dataset --save_dir dataset/labelImg_format
  • --data_dir: COCO格式数据集所在目录。示例为dataset/COCO_dataset
  • --save_dir: 保存转换后的数据集目录。默认为dataset/COCO_dataset_labelImg

转换后结构如下： 

  labelImg_dataset
  ├── train
  │   ├── 000000000001.jpg
  │   ├── 000000000001.txt
  │   ├── 000000000002.jpg
  │   └── classes.txt
  └── val
      ├── 000000000001.jpg
      ├── 000000000001.txt
      └── classes.txt

labelImg可视化 

  $ cd dataset/labelImg_dataset
$ labelImg train train/classes.txt

# or
$ labelImg val val/classes.txt
Edit this page

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