augmix#

Modules

composer.algorithms.augmix.augmix

Core AugMix classes and functions.

AugMix (Hendrycks et al, 2020) creates multiple independent realizations of sequences of image augmentations, applies each sequence with random intensity, and returns a convex combination of the augmented images and the original image.

See the Method Card for more details.

Functions

augmix_image

Applies the AugMix (Hendrycks et al, 2020) data augmentation.

Classes

AugMix

The AugMix data augmentation technique.

AugmentAndMixTransform

Wrapper module for augmix_image() that can be passed to torchvision.transforms.Compose.