mixmo.loaders.dataset_wrapper

Dataset wrappers for multi-input multi-output models with data augmentation

Classes

DADataset(dataset, num_classes, num_members, …)	Dataset wrapper with with outputs formatted as dictionaries and AugMix augmentation
MSDADataset(dataset, num_classes, …)	Dataset wrapper that returns dictionaries and applies MSDA augmentations
MixMoDataset(dataset, num_classes, …)	Dataset wrapper that returns dictionaries of multiple samples, and applies MSDA augmentations