Struct ImageAugmenter 

pub struct ImageAugmenter { /* private fields */ }

Image augmenter for data augmentation

Implementations

impl ImageAugmenter

pub fn new() -> Self

Create a new image augmenter

pub fn with_rotation_range(self, range: (Float, Float)) -> Self

Set rotation range in degrees

pub fn with_zoom_range(self, range: (Float, Float)) -> Self

Set zoom range as factors

pub fn with_width_shift_range(self, range: Float) -> Self

Set width shift range as fraction

pub fn with_height_shift_range(self, range: Float) -> Self

Set height shift range as fraction

pub fn with_horizontal_flip(self, enabled: bool) -> Self

Enable horizontal flipping

pub fn with_vertical_flip(self, enabled: bool) -> Self

Enable vertical flipping

pub fn with_brightness_range(self, range: (Float, Float)) -> Self

Set brightness adjustment range

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

Set random seed for reproducibility

Trait Implementations

impl Clone for ImageAugmenter

fn clone(&self) -> ImageAugmenter

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ImageAugmenter

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

Formats the value using the given formatter.

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 3]>>> for ImageAugmenter

fn transform(&self, x: &Array3<Float>) -> Result<Array3<Float>>

Transform the input data