Trait Model 

pub trait Model: Send + Sync {
    // Required methods
    fn metadata(&self) -> &ModelMetadata;
    fn predict(&mut self, input: &RasterBuffer) -> Result<RasterBuffer>;
    fn predict_batch(
        &mut self,
        inputs: &[RasterBuffer],
    ) -> Result<Vec<RasterBuffer>>;
    fn input_shape(&self) -> (usize, usize, usize);
    fn output_shape(&self) -> (usize, usize, usize);
}

Trait for ML models

Required Methods

fn metadata(&self) -> &ModelMetadata

Returns the model metadata

fn predict(&mut self, input: &RasterBuffer) -> Result<RasterBuffer>

Predicts on a single raster buffer

Errors

Returns an error if prediction fails

fn predict_batch( &mut self, inputs: &[RasterBuffer], ) -> Result<Vec<RasterBuffer>>

Predicts on multiple raster buffers (batch prediction)

Errors

Returns an error if prediction fails

fn input_shape(&self) -> (usize, usize, usize)

Returns the expected input shape (channels, height, width)

fn output_shape(&self) -> (usize, usize, usize)

Returns the expected output shape (channels, height, width)

Implementors

impl Model for OnnxModel