Struct DocOrientationClassifierBuilder 

pub struct DocOrientationClassifierBuilder { /* private fields */ }

Builder for document orientation classifier

This struct provides a builder pattern for creating a document orientation classifier with custom configuration options.

Implementations

impl DocOrientationClassifierBuilder

pub fn new() -> Self

Creates a new document orientation classifier builder

Initializes a new instance of the document orientation classifier builder with default configuration options.

Returns

A new instance of DocOrientationClassifierBuilder

pub fn model_path(self, model_path: impl Into<PathBuf>) -> Self

Sets the model path for the classifier

Specifies the path to the ONNX model file that will be used for inference.

Arguments

• model_path - Path to the ONNX model file

Returns

The updated builder instance

pub fn model_name(self, model_name: impl Into<String>) -> Self

Sets the model name for the classifier

Specifies the name of the model being used.

Arguments

• model_name - Name of the model

Returns

The updated builder instance

pub fn batch_size(self, batch_size: usize) -> Self

Sets the batch size for the classifier

Specifies the number of images to process in each batch.

Arguments

• batch_size - Number of images to process in each batch

Returns

The updated builder instance

pub fn enable_logging(self, enable: bool) -> Self

Enables or disables logging for the classifier

Controls whether logging is enabled during classification.

Arguments

• enable - Whether to enable logging

Returns

The updated builder instance

pub fn ort_session(self, config: OrtSessionConfig) -> Self

Sets the ONNX Runtime session configuration

This function sets the ONNX Runtime session configuration for the predictor.

pub fn session_pool_size(self, size: usize) -> Self

Sets the session pool size for concurrent predictions

This function sets the size of the session pool used for concurrent predictions. The pool size must be >= 1.

Arguments

• size - The session pool size (minimum 1)

Returns

The updated builder instance

pub fn topk(self, topk: usize) -> Self

Sets the number of top predictions to return

Specifies how many of the top predictions to return for each image.

Arguments

• topk - Number of top predictions to return

Returns

The updated builder instance

pub fn input_shape(self, input_shape: (u32, u32)) -> Self

Sets the input shape for the model

Specifies the input shape (width, height) that the model expects.

Arguments

• input_shape - Input shape as (width, height)

Returns

The updated builder instance

pub fn build( self, model_path: &Path, ) -> Result<DocOrientationClassifier, OCRError>

Builds the document orientation classifier

Creates a new instance of the document orientation classifier with the configured options.

Arguments

• model_path - Path to the ONNX model file

Returns

A new instance of DocOrientationClassifier or an error if building fails

Trait Implementations

impl Default for DocOrientationClassifierBuilder

fn default() -> Self

Creates a new document orientation classifier builder with default settings

This is equivalent to calling DocOrientationClassifierBuilder::new().

Returns

A new instance of DocOrientationClassifierBuilder with default settings