bencevans / pycococt Goto Github PK

View Code? Open in Web Editor NEW

COCO-CameraTrap Indexer. An extension to pycocotools including location and sequence indexing.

License: MIT License

Python 100.00%

pycococt's Introduction

pycococt

COCO-CameraTrap Indexer. An extension to pycocotools including location and sequence indexing.

Features

Extends / Backwards compatable with pycocotools
Indexing for Locations and Sequences

Install

pip install pycococt

Usage

from pycococt.cococt import COCOCT

ds = COCOCT('path to COCO-CameraTrap')

# all the core indexes and helpers in pycocotools
ds.imgs
ds.anns
ds.cats
ds.catToImgs
ds.imgToAnns

# with aditional camera trap indexes
ds.locs      # = [
             #   'Location1',
             #   'Location2',
             #   'Location3',
             #   ...
             # ]

ds.locToImgs # = { 'Location1': {
             #        'imageid': {
             #           'id': 'imageid',
             #           'location': 'Location1',
             #           'height': ...
             #         }
             #      },
             #      ...
             #    }

ds.locToSeqs # = {
             #     'Location1': [
             #         'seq id 1_1',
             #         'seq id 1_2',
             #         ...
             #     ],
             #     'Location2': [
             #         'seq id 2_1',
             #         'seq id 2_2',
             #         ...
             #     ]
             #     ...
             # }

ds.seqToImgs # = { 'sequence 1 id': [
             #       {
             #           'id': 'image id',
             #           ...
             #        },
             #        {
             #           'id': 'image id',
             #           ...
             #        },
             #        ...
             #     ],
             #     'sequence 2 id: [
             #       ...
             #     ]
             #       ...
             #    }