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 a single image or batch of images.

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.