A CLI tool can create a specific task-dataset you want based on COCO dataset.
Given the annotation JSON file, this tool will help you download the data and set the symbolic links from data_dir to task_dir !!
$ pip3 install pycocotools requests click$ git clone [email protected]:tw-yshuang/coco2yolo.git
# or
$ git clone https://github.com/tw-yshuang/coco2yolo.git$ chmod +x coco2yolo # add execute permission on UNIX/Linux
$ ./coco2yolo [OPTIONS] [CAT_INFOS]
# or
$ python3 coco2yolo [OPTIONS] [CAT_INFOS]Usage: coco2yolo [OPTIONS] [CAT_INFOS]...
Options:
--help Show this message and exit.
-ann-path, --annotation-path TEXT
JSON file. Path for label. [required]
-img-dir, --image-download-dir TEXT
The directory of the image data place.
-task-dir, --task-categories-dir TEXT
Build a directory that follows the task-required categories.
-cat-t, --category-type TEXT Category input type. (interactive | file) [default: interactive]
-set, --set-computing-type TEXT
Set Computing for the data. (union | intersection) [default: union]
It is a must option to execute this tool:
$ ./coco2yolo -ann-path <annotations/instances_xxx.json>Choose the category input type you want:
$ ./coco2yolo -ann-path <annotations/instances_xxx.json> -cat-t <interactive | file>interactive : type the categories name you wish. (default)
file: locate a categories JSON file, e.g.
# example.json
["person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck"]Choose the set computing type you want:
$ ./coco2yolo -ann-path <annotations/instances_xxx.json> -set <union | intersection>union: collect the data that contain the categories you set from COCO images. (default)
interesction: collect the data that only satisfy the categories you set from COCO images.
It will automatically create the directory that you given.
If you do not have the COCO images data in -img-dir <img_dir>, this tool will automatically download the images data to <img_dir> by task.
In the <task_dir> the images data will set the symbolic links from <img_dir> to <task_dir>
$ ./coco2yolo -ann-path ./annotations/instances_train2017.json
...
Enter the directory that you save COCO images:
Enter the directory that you want to save the task:$ ./coco2yolo -ann-path ./annotations/instances_train2017.json -img-dir ./train2017 -task-dir ./task -set intersection
...
Enter the categories name you wish(split: ', '):$ ./coco2yolo -ann-path ./annotations/instances_train2017.json -img-dir ./train2017 -task-dir ./task -cat-t file -set intersection
...
Enter the category JSON file you want:
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent