dgs.models.dataset.alphapose

Load bboxes and poses from existing .json file, generated by AP.

By default, AlphaPose saves the results for all images in one json file, which is similar to the output box_format used by COCO. For the basic AlphaPose output box_format see https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/output.md.

Within the .json files, AlphaPose uses the following overall structure:

image_id

The name of the image as string. No additional path information is given.

keypoints

The body part locations and detection confidence formatted as array [x1, y1, c1, x2, y2, c2, ...]. Where c is the confidence score in the range [0,1] for the MPII dataset and range [0,6] for the COCO dataset.

There can be an arbitrary number of key points, it is just known, that the length has to be divisible by three.

score

The confidence score for the whole person, computed by AlphaPoses’ parametric pose NMS.

box

The detected bounding box as an array of floats in XYWH box_format.

idx

The integer index of the detected person.

Module Classes

AlphaPoseLoader(*args, **kwargs)

Load precomputed json files.