unipose.datasets.ConcatJointDataset

class documentation

class ConcatJointDataset(ConcatDataset):

A dataset that concatenates multiple datasets.

Parameters
datasets	The datasets to concatenate.

Method	`__init__`	Undocumented
Method	`make_dataloader`	Make a dataloader for the concat-ed dataset.

def __init__(self, datasets): ¶

Undocumented

def make_dataloader(self, image_size=256, scale_factor=4, *args, **kwargs): ¶

Make a dataloader for the concat-ed dataset.

A unipose dataset returns a dictionary with the following keys: "images", "keypoint_images", "masks", "extra_keypoints", "extra_tokens".