Skip to content

ashnair1/COCO-Assistant

Repository files navigation

COCO-Assistant

CircleCI Codacy Badge Code style: black PyPi License License: MIT

Note: Please install from source. I've encountered some issues with uploading to Pypi and will look into this when I have time.

Helper for dealing with MS-COCO annotations.

Overview

The MS COCO annotation format along with the pycocotools library is quite popular among the computer vision community. Yet I for one found it difficult to play around with the annotations. Deleting a specific category, combining multiple mini datasets to generate a larger dataset, viewing distribution of classes in the annotation file are things I would like to do without writing a separate script for each scenario.

The COCO Assistant is designed (or being designed) to assist with this problem. Please note that currently, the Assistant can only help out with object detection datasets. Any contributions and/or suggestions are welcome.

Package features

COCO-Assistant currently supports the following features:

  • Merge datasets.
  • Remove specfiic category from dataset.
  • Generate annotations statistics - distribution of object areas and category distribution.
  • Annotation visualiser for viewing the entire dataset.

Documentation

Project documentation can be found here

Credits

This package was created with the zillionare/cookiecutter-pypackage project template.