Struct ContrastiveAugmentationBuilder 

pub struct ContrastiveAugmentationBuilder { /* private fields */ }

Builder for ContrastiveAugmentation.

Implementations

impl ContrastiveAugmentationBuilder

pub fn new() -> Self

Create a new builder with default config.

pub fn crop_scale(self, min: f64, max: f64) -> Self

Set the crop scale range.

pub fn aspect_ratio(self, min: f64, max: f64) -> Self

Set the aspect ratio range.

pub fn horizontal_flip_prob(self, prob: f64) -> Self

Set the horizontal flip probability.

pub fn color_jitter( self, brightness: f64, contrast: f64, saturation: f64, hue: f64, ) -> Self

Set the color jitter parameters.

pub fn color_jitter_prob(self, prob: f64) -> Self

Set the color jitter probability.

pub fn grayscale_prob(self, prob: f64) -> Self

Set the grayscale probability.

pub fn gaussian_blur(self, kernel_size: u32, sigma_range: (f64, f64)) -> Self

Enable Gaussian blur with the specified kernel size.

pub fn blur_prob(self, prob: f64) -> Self

Set the blur probability.

pub fn output_size(self, width: u32, height: u32) -> Self

Set the output size.

pub fn seed(self, seed: u64) -> Self

Set a fixed random seed for reproducibility.

pub fn build(self) -> ContrastiveAugmentation

Build the ContrastiveAugmentation instance.

Trait Implementations

impl Clone for ContrastiveAugmentationBuilder

fn clone(&self) -> ContrastiveAugmentationBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ContrastiveAugmentationBuilder

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

Formats the value using the given formatter.

impl Default for ContrastiveAugmentationBuilder

fn default() -> Self

Returns the “default value” for a type.