Struct ContrastiveAugmentation 

pub struct ContrastiveAugmentation { /* private fields */ }

SimCLR-style contrastive augmentation pipeline.

Example

use ruvector_cnn::contrastive::ContrastiveAugmentation;

let aug = ContrastiveAugmentation::builder()
    .crop_scale(0.08, 1.0)
    .horizontal_flip_prob(0.5)
    .color_jitter(0.4, 0.4, 0.4, 0.1)
    .output_size(224, 224)
    .build();

// Generate two augmented views of an image (requires augmentation feature)
// let (view1, view2) = aug.generate_pair(&image)?;

Implementations

impl ContrastiveAugmentation

pub fn builder() -> ContrastiveAugmentationBuilder

Create a builder for ContrastiveAugmentation.

pub fn config(&self) -> &AugmentationConfig

Get the current configuration.

Trait Implementations

impl Clone for ContrastiveAugmentation

fn clone(&self) -> ContrastiveAugmentation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ContrastiveAugmentation

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

Formats the value using the given formatter.

impl Default for ContrastiveAugmentation

fn default() -> Self

Returns the “default value” for a type.