mixmo.augmentations.augmix

Implementation following setups from PuzzleMix authors According to the seminal code: https://github.com/google-research/augmix/blob/master/cifar.py This code structure is borrowed from: https://github.com/ildoonet/pytorch-randaugment/blob/616ef12a5176169b4e1e645728f3dedd1a5a148e/RandAugment/augmentations.py

Functions

AutoContrast(img, *args, **kwargs)
Brightness(img, v, **kwargs)
Color(img, v, **kwargs)
Contrast(img, v, **kwargs)
Equalize(img, *args, **kwargs)
Identity(img, v, **kwargs)
Invert(img, *args, **kwargs)
Posterize(img, v, **kwargs)
Rotate(img, v[, myrandom])
Sharpness(img, v, **kwargs)
ShearX(img, v[, myrandom])
ShearY(img, v[, myrandom])
Solarize(img, v, **kwargs)
TranslateX(img, v[, myrandom])
TranslateY(img, v[, myrandom])
augment_list([include_auto_contrast])
get_value_when_none(value, default_value)

Classes

AugMix([seed, mixture_depth, mixture_width, …])