dgs.models.submission.posetrack21

Module for creating submission files for PoseTrack21 .

References

https://github.com/anDoer/PoseTrack21/blob/main/doc/dataset_structure.md

https://github.com/leonid-pishchulin/poseval

Notes

The structure of the PT21 submission file is similar to the structure of the inputs:

{
    "images": [
        {
            "file_name": "images/train/000001_bonn_train/000000.jpg",
            "id": 10000010000,
            "frame_id": 10000010000
        },
    ],
    "annotations": [
        {
            "bbox": [x1,  y1, w, h],
            "image_id": 10000010000,
            "keypoints": [x1, y1, vis1, ..., x17, y17, vis17],
            "scores": [s1, ..., s17],
            "person_id": 1024,
            "track_id": 0
        },
    ]
}

Additionally, note that the visibilities are ignored during evaluation.

Module Classes

PoseTrack21Submission(config, path)

Class for creating and appending to a PoseTrack21 -style submission file.