dgs.models.dataset.posetrack21.PoseTrack21Torchreid

class dgs.models.dataset.posetrack21.PoseTrack21Torchreid(*args: Any, **kwargs: Any)[source]

Load PoseTrack21 as torchreid dataset. Depending on the argument instance this Dataset either contains image crops or key point crops.

Reference

Doering et al. Posetrack21: A dataset for person search, multi-object tracking and multi-person pose tracking. IEEE / CVF 2022.

URL

https://github.com/andoer/PoseTrack21

Dataset statistics

  • identities: The training set contains 5474 unique person ids. The biggest person id is 6878

  • images: 163411 images, divided into: 96215 train, 46751 test (gallery), and 20444 val (query)

param root:

Root directory of all the datasets. Default “./data/”.

type root:

str

param instance:

Whether this module works as a TorchreidImageDataset or a custom TorchreidPoseDataset. Has to be one of: [“images”, “key_points”]. Default “all”.

type instance:

str

Notes

The bbox crops are generated using either the modified self.download_dataset() or if you don’t want to use default configuration something similar using extract_all_bboxes().

Notes

Train is for training the model. The query and gallery are used for testing, where for each image in the query you find similar persons in the gallery set.

__init__(root: str = '', instance: str = 'images', **kwargs)[source]

Methods

download_dataset([dataset_dir, dataset_url])

Originally intended to download the dataset, but authentication is required.

process_file(filepath, crops_dir, *[, ...])

Process all the data in a single directory.

show_summary()

Print dataset summary.

Attributes

dataset_dir

Name of the directory containing the dataset within root.