coco2yolo

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 !!

Installation

Dependency Packages

$ pip3 install pycocotools requests click

Clon the Repository

$ git clone [email protected]:tw-yshuang/coco2yolo.git
  # or
$ git clone https://github.com/tw-yshuang/coco2yolo.git

Execute

$ chmod +x coco2yolo # add execute permission on UNIX/Linux
$ ./coco2yolo [OPTIONS] [CAT_INFOS]
  # or
$ python3 coco2yolo  [OPTIONS] [CAT_INFOS]

Usage

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]

Locate The Annotation

It is a must option to execute this tool:

$ ./coco2yolo -ann-path <annotations/instances_xxx.json>

Category Type

Choose the category input type you want:

$ ./coco2yolo -ann-path <annotations/instances_xxx.json> -cat-t <interactive | file>

Mode

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"]

Set Computing Type

Choose the set computing type you want:

$ ./coco2yolo -ann-path <annotations/instances_xxx.json> -set <union | intersection>

Mode

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.

Note

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>

Example

$ ./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: