mixmo.loaders.dataset_wrapper.MSDADataset¶

class mixmo.loaders.dataset_wrapper.MSDADataset(dataset, num_classes, num_members, dict_config, properties)[source]¶

Dataset wrapper that returns dictionaries and applies MSDA augmentations

__init__(dataset, num_classes, num_members, dict_config, properties)¶: Initialize self. See help(type(self)) for accurate signature.

Methods

`__init__`(dataset, num_classes, num_members, …)	Initialize self.
`call_dataset`(index[, seed])	Get target and image, apply AugMix if necessary and return dictionary
`call_msda`(index_0[, mixmo_mask, seed_da])	Get two samples and mix them.
`set_ratio_epoch`(ratioepoch)

Attributes

reverse_if_first_minor

call_msda(index_0, mixmo_mask=None, seed_da=None)[source]¶: Get two samples and mix them. Return a dictionary of sample and label