简介

labelImg标注的数据集格式转换为YOLO格式。

labelImg结构如下:

  labelImg_dataset
├── classes.txt
├── images(13).jpg
├── images(13).txt
├── images(3).jpg
├── images(3).txt
├── images4.jpg
├── images4.txt
├── images5.jpg
├── images5.txt
├── images6.jpg
├── images7.jpg
└── images7.txt
  

转换

  labelImg_to_yolov5 --data_dir dataset/labelImg_dataset \
                   --save_dir dataset/labelImg_dataset_output \
                   --val_ratio 0.2 \
                   --have_test \
                   --test_ratio 0.2
  
  • --data_dir: labelme标注的数据所在路径,示例为dataset/labelImg_dataset
  • --save_dir: 转换后数据存储路径,默认为dataset/labelImg_dataset_publaynet
  • --val_ratio: 验证集所占比例,默认为总量的0.2
  • --have_test: 是否划出测试集,默认为False,如果想要划分测试集,直接加上该参数即可。
  • --test_ratio: 测试集的比例,默认为总量的0.2

转换后结构如下:

  labelImg_dataset_output/
├── classes.txt
├── images
│   ├── images(13).jpg
│   ├── images(3).jpg
│   ├── images4.jpg
│   ├── images5.jpg
│   └── images7.jpg
├── labels
│   ├── images(13).txt
│   ├── images(3).txt
│   ├── images4.txt
│   ├── images5.txt
│   └── images7.txt
├── non_labels        # 这个是没有标签的图像目录
│   └── images6.jpg
├── test.txt
├── train.txt
└── val.txt
  

进一步转换为COCO格式

  yolov5_to_coco --data_dir dataset/labellImg_dataset_output
  

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