Enum AugOp 

pub enum AugOp {
    RandomCrop {
        crop_size: usize,
    },
    CenterCrop {
        crop_size: usize,
    },
    HorizontalFlip {
        prob: f32,
    },
    Resize {
        target: usize,
    },
    ColorJitter {
        brightness: f32,
        contrast: f32,
        saturation: f32,
    },
    RandomGrayscale {
        prob: f32,
    },
    Normalize {
        mean: [f32; 3],
        std: [f32; 3],
    },
}

Enum-dispatched augmentation operations — no dyn Trait, no heap boxing.

Each variant carries the hyperparameters it needs. Stochastic operations receive a mutable LcgRng reference at call time via AugOp::apply.

Variants

RandomCrop

Randomly crop to [channels, crop_size, crop_size].

Fields

crop_size: usize

CenterCrop

Deterministic centre crop to [channels, crop_size, crop_size].

Fields

crop_size: usize

HorizontalFlip

Randomly flip the image horizontally with the given probability.

Fields

prob: f32

Resize

Bilinear resize to [channels, target, target].

Fields

target: usize

ColorJitter

Colour jitter: brightness, contrast, saturation perturbation magnitudes.

Fields

brightness: f32

contrast: f32

saturation: f32

RandomGrayscale

Convert to grayscale with the given probability (RGB images only).

Fields

prob: f32

Normalize

Channel-wise normalisation: (x - mean[c]) / std[c].

Fields

mean: [f32; 3]

std: [f32; 3]

Implementations

impl AugOp

pub fn apply( &self, img: &[f32], channels: usize, h: usize, w: usize, rng: &mut LcgRng, ) -> VisionResult<(Vec<f32>, usize, usize)>

Apply this augmentation to a CHW image.

Parameters

• img: flat [channels × h × w] input buffer.
• channels: number of channels (e.g., 3 for RGB).
• h, w: spatial height and width of the input image.
• rng: mutable RNG for stochastic operations; deterministic ops ignore it.

Returns

(new_img, new_h, new_w) — the transformed image and its (possibly updated) spatial dimensions. channels is never changed.

Errors

Propagates errors from the underlying operation functions (invalid dimensions, mismatched buffers, non-positive std, etc.).

Trait Implementations

impl Clone for AugOp

fn clone(&self) -> AugOp

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AugOp

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.