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 git@github.com: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: