keypoints_dataset

Description

MS COCO Keypoints dataset handle with subsetting functionality and annotation transformations.

Classes

COCOBoxToSegMask

Turn the bounding box info from COCO annotations into a single binary segmentation mask.

COCOSegToSegMask

Turn COCO instance segmentation annotations into a single binary segmentation mask.

KeypointsDataset

Handler for (a subset of) a COCO keypoints dataset.

PadAndScaleAnnotatedImg

Pickleable wrapper to pad and scale both an image and its COCO annotation to a target size.