oxicuda_ssl/augment/mod.rs
1//! Standard SSL data augmentation helpers operating on `[C, H, W]` CHW tensors.
2
3pub mod color;
4pub mod multi_crop;
5pub mod rand_augment;
6pub mod solarize_blur;
7
8pub use rand_augment::{
9 AugOp, AutoAugPolicy, AutoAugmentConfig, RandAugmentConfig, SubPolicy, all_aug_ops,
10 apply_aug_op, auto_augment, rand_augment,
11};
12pub use solarize_blur::{
13 SimClrBlurSolarConfig, add_gaussian_noise, gaussian_blur_chw, random_gaussian_blur_chw,
14 random_solarize, simclr_blur_solar, solarize,
15};