Struct SuperResolution 

pub struct SuperResolution { /* private fields */ }

Super-resolution model using ONNX Runtime

Implementations

impl SuperResolution

pub fn from_file<P: AsRef<Path>>( path: P, config: SuperResConfig, ) -> Result<Self>

Load a super-resolution model from an ONNX file

Arguments

• path - Path to the ONNX model file
• config - Super-resolution configuration

Errors

Returns an error if the model file cannot be loaded or is invalid

Example

use oxigdal_ml::superres::{SuperResolution, SuperResConfig};

let config = SuperResConfig::default();
let model = SuperResolution::from_file("real_esrgan_2x.onnx", config)?;

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

Upscale a raster using the super-resolution model

Arguments

• input - Input raster buffer

Errors

Returns an error if inference fails or input validation fails

Example

use oxigdal_ml::superres::{SuperResolution, SuperResConfig};
use oxigdal_core::buffer::RasterBuffer;
use oxigdal_core::types::RasterDataType;

let config = SuperResConfig::default();
let mut model = SuperResolution::from_file("real_esrgan_2x.onnx", config)?;

let input = RasterBuffer::zeros(512, 512, RasterDataType::Float32);
let output = model.upscale(&input)?;