Module augment 

Standard SSL data augmentation helpers operating on [C, H, W] CHW tensors.

Re-exports

pub use rand_augment::AugOp;

pub use rand_augment::AutoAugPolicy;

pub use rand_augment::AutoAugmentConfig;

pub use rand_augment::RandAugmentConfig;

pub use rand_augment::SubPolicy;

pub use rand_augment::all_aug_ops;

pub use rand_augment::apply_aug_op;

pub use rand_augment::auto_augment;

pub use rand_augment::rand_augment;

pub use solarize_blur::SimClrBlurSolarConfig;

pub use solarize_blur::add_gaussian_noise;

pub use solarize_blur::gaussian_blur_chw;

pub use solarize_blur::random_gaussian_blur_chw;

pub use solarize_blur::random_solarize;

pub use solarize_blur::simclr_blur_solar;

pub use solarize_blur::solarize;

Modules

color

Color augmentation primitives operating on [C, H, W] (CHW) RGB images.

multi_crop

Multi-crop augmentation strategy used by SwAV and DINO.

rand_augment

RandAugment and AutoAugment augmentation policies for CHW images.

solarize_blur

Gaussian blur, solarization, and Gaussian-noise augmentations for CHW images.