composer.algorithms.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 AugMix (Hendrycks et al, 2020) data augmentation to an image.

Classes

AugMix

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

AugmentAndMixTransform

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