跳转至

labelImg → PubLayNet

简介

将 labelImg 格式数据集转换为 PubLayNet 格式。

labelImg 结构如下

Tip

具体结构示例文件，可移步：labelImg_dataset

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
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

转换

1
2
3
4
labelImg_to_publaynet --data_dir dataset/labelImg_dataset \
                       --val_ratio 0.2 \
                       --have_test \
                       --test_ratio 0.2
  • --data_dir: COCO 格式数据集所在目录。示例为 dataset/labelImg_dataset
  • --save_dir: 保存转换后的数据集目录。默认为 dataset/labelImg_dataset_publaynet
  • --val_ratio: 验证集数目占数据集总数比例，默认为 0.2.
  • --have_test: 是否有测试集。默认为 False，如果出现，则为 True
  • --test_ratio: 测试集数目占数据集总数比例，默认为 0.2

转换后结构如下

Tip

具体结构示例文件，可移步：publaynet_dataset

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
publaynet_dataset
├── test
│   ├── images5.jpg
│   └── images5.txt
├── test.json
├── train
│   ├── images(13).jpg
│   ├── images(13).txt
│   ├── images(3).jpg
│   ├── images(3).txt
│   ├── images4.jpg
│   ├── images4.txt
│   ├── images5.jpg
│   ├── images5.txt
│   ├── images7.jpg
│   └── images7.txt
├── train.json
├── val
│   ├── images(13).jpg
│   ├── images(13).txt
│   ├── images5.jpg
│   ├── images5.txt
│   ├── images7.jpg
│   └── images7.txt
└── val.json

评论