mml.core.data_loading.augmentations.augmentation_module

class AugmentationModule[source]

Bases: ABC

__init__(device: str, cfg: ListConfig | List[Dict[str, Any]], is_first: bool, is_last: bool, means: RGBInfo | None, stds: RGBInfo | None)[source]

static get_data_format(inpt: Any) → DataFormat[source]: Checks the input data format.

class AugmentationModuleContainer[source]

Bases: object

__init__(modules: List[AugmentationModule])[source]

class DataFormat[source]

Bases: StrEnum

An enumeration.

BATCHED_SAMPLE_DICTS = 'batched_sample_dicts'

MULTI_TASK_SAMPLE_DICTS = 'multitask_sample_dicts'

SINGLE_SAMPLE_DICT = 'single_sample_dict'

__new__(value)