Struct ImagePreprocessor 

pub struct ImagePreprocessor { /* private fields */ }

Builder for an image preprocessing pipeline.

See the module-level docs for the intended flow. An instance is cheap to clone so a single pre-built preprocessor can be shared across threads.

Implementations

impl ImagePreprocessor

pub fn new(target_width: u32, target_height: u32) -> Self

Create a new preprocessor for a given output size.

pub fn with_pixel_layout(self, layout: PixelLayout) -> Self

Set the input pixel layout. Default is PixelLayout::Rgb.

pub fn with_tensor_layout(self, layout: TensorLayout) -> Self

Set the output tensor layout. Default is TensorLayout::Nchw.

pub fn with_input_range(self, range: InputRange) -> Self

Set the scalar range of the source image. Default is InputRange::U8.

pub fn with_mean(self, mean: [f32; 3]) -> Self

Set the per-channel normalisation mean.

pub fn with_std(self, std: [f32; 3]) -> Self

Set the per-channel normalisation std-dev.

pub fn with_imagenet_normalization(self) -> Self

Apply ImageNet mean [0.485, 0.456, 0.406] and std [0.229, 0.224, 0.225].

pub fn target_width(&self) -> u32

Target width in pixels.

pub fn target_height(&self) -> u32

Target height in pixels.

pub fn process_u8_rgb( &self, pixels: &[u8], src_w: u32, src_h: u32, ) -> MlResult<Vec<f32>>

Process a raw u8 RGB(3-channel) buffer of size src_w * src_h * 3.

The output is a flattened Vec<f32> containing a single image (no batch dimension). Call ImagePreprocessor::batch_shape to learn the logical shape.

Errors

Returns MlError::Preprocess if:

• pixels.len() != src_w * src_h * 3,
• either dimension (source or target) is zero.

pub fn batch_shape(&self) -> Vec<usize>

Return the logical shape of the output tensor with a leading batch dim of 1. Matches the flat buffer returned by ImagePreprocessor::process_u8_rgb.

Trait Implementations

impl Clone for ImagePreprocessor

fn clone(&self) -> ImagePreprocessor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ImagePreprocessor

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

Formats the value using the given formatter.