unipose.datasets.AnimalKingdomDataset

class documentation

class AnimalKingdomDataset(BaseJointDataset):

View In Hierarchy

Undocumented

Method __getitem__ Undocumented
Method __init__ Undocumented
Method __len__ Undocumented
Constant EXTRA_TOKENS Undocumented
Constant MAPPING Undocumented
Instance Variable animal_kingdom Undocumented
Instance Variable ids Undocumented
Instance Variable json Undocumented
Instance Variable path Undocumented
Instance Variable split Undocumented
Instance Variable sub_category Undocumented

Inherited from BaseJointDataset:

Method make_dataloader Make a dataloader for the dataset.
Method _apply_mapping Apply the mapping from any dataset to the unipose format. This is used to reorder the keypoints and mask to the unipose format, according to the MAPPING and EXTRA_TOKENS class variables.
def __getitem__(self, idx):
overrides unipose.datasets.BaseJointDataset.__getitem__

Undocumented

def __init__(self, path, sub_category='ak_P1', split='train'):
overrides unipose.datasets.BaseJointDataset.__init__

Undocumented

Parameters
path:strUndocumented
sub_category:Literal['ak_P1', 'ak_P2', 'ak_P3_amphibian', 'ak_P3_bird', 'ak_P3_fish', 'ak_P3_mammal', 'ak_P3_reptile']Undocumented
split:Literal['train', 'test']Undocumented
def __len__(self):
overrides unipose.datasets.BaseJointDataset.__len__

Undocumented

EXTRA_TOKENS: dict[str, int] =
overrides unipose.datasets.BaseJointDataset.EXTRA_TOKENS

Undocumented

Value
{'head_top': 0,
 'left_eye': 1,
 'right_eye': 2,
 'mouth_left': 4,
 'mouth_right': 5,
 'mouth_bottom': 6,
 'torso_middle': 13,
...
MAPPING: dict[int, int] =
overrides unipose.datasets.BaseJointDataset.MAPPING

Undocumented

Value
{0: 3,
 1: 7,
 2: 9,
 3: 11,
 4: 8,
 5: 10,
 6: 12,
...
animal_kingdom =

Undocumented

ids =

Undocumented

json =

Undocumented

path =

Undocumented

split =

Undocumented

sub_category =

Undocumented