Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

Args: index (int): Index Returns: tuple: (sample, target) where target is class_index of the target class.

Retrieve the item at the given index. Args: index (int): The index of the item to be retrieved.

Retrieve the item at the given index. Args: index (int): The index of the item to be retrieved.

Wrapper around the pyrender renderer to render SMPL meshes. Args: cfg (CfgNode): Model config file. faces (np

Object used to render 3D keypoints. Faster for use during training. Args: cfg (CfgNode): Model config file.

Extension of the official SMPL implementation to support more joints. Args: Same as SMPLLayer. joint_regresso

Setup HMR2 model Args: cfg (CfgNode): Config file as a yacs CfgNode

Performs masking on a dict of modalities (both image based and sequence based modalities) Args: modality_info: Dict with the moda

Performs masking for transfer on a dict of modalities (both image based and sequence based modalities), by specifying which modalities are inp

Unified data augmentation for FourM Args: transforms_dict (dict): Dict of transforms for each modality image_augmente

Metadata transform that takes in a metadata dictionary and converts it into a string, or list of strings (for chunked span masking).

Dataset wrapper that repeats the original dataset n times. Args: original_dataset (torch.utils.data.Dataset): The origin

Dataset wrapper that randomly subsamples the original dataset. Args: original_dataset (torch.utils.data.Dataset): The or