Package com.itextpdf.pdfocr.onnxtr.orientation

Class OnnxOrientationPredictorProperties

java.lang.Object

com.itextpdf.pdfocr.onnxtr.orientation.OnnxOrientationPredictorProperties

       public class OnnxOrientationPredictorProperties extends Object
      

Properties for configuring crop orientation ONNX models.

It contains a path to the model, model input properties and a model output mapper.

Constructor Summary

Constructors

Constructor	Description
OnnxOrientationPredictorProperties(String modelPath, OnnxInputProperties inputProperties, IOutputLabelMapper<TextOrientation> outputMapper)	Creates new crop orientation predictor properties.

Method Summary

Modifier and Type	Method	Description
boolean	equals(Object o)
OnnxInputProperties	getInputProperties()	Returns the ONNX model input properties.
String	getModelPath()	Returns the path to the ONNX model.
IOutputLabelMapper<TextOrientation>	getOutputMapper()	Returns the ONNX model output mapper.
int	hashCode()
static OnnxOrientationPredictorProperties	mobileNetV3(String modelPath)	Creates a new crop orientation properties object for existing pre-trained MobileNetV3 models, stored on disk.
String	toString()

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Details

OnnxOrientationPredictorProperties

             public OnnxOrientationPredictorProperties
             (String modelPath, OnnxInputProperties inputProperties, IOutputLabelMapper<TextOrientation> outputMapper)
            

Creates new crop orientation predictor properties.

Parameters:

modelPath - path to the ONNX model to load

inputProperties - ONNX model input properties

outputMapper - ONNX model output mapper

Method Details

mobileNetV3

             public static OnnxOrientationPredictorProperties mobileNetV3
             (String modelPath)
            

Creates a new crop orientation properties object for existing pre-trained MobileNetV3 models, stored on disk. This is the only crop orientation model architecture available in OnnxTR.

This can be used to load the following models from OnnxTR:

• mobilenet_v3_small_crop_orientation
• mobilenet_v3_small_crop_orientation (8-bit quantized)

Parameters:

modelPath - path to the pre-trained model

Returns:

a new crop orientation properties object for a MobileNetV3 model

getModelPath

             public String getModelPath()
            

Returns the path to the ONNX model.

Returns:

the path to the ONNX model

getInputProperties

             public OnnxInputProperties getInputProperties()
            

Returns the ONNX model input properties.

Returns:

the ONNX model input properties

getOutputMapper

             public IOutputLabelMapper<TextOrientation> getOutputMapper()
            

Returns the ONNX model output mapper.

Returns:

the ONNX model output mapper

hashCode

             public int hashCode()
            

Overrides:

hashCode in class Object

equals

             public boolean equals
             (Object o)
            

Overrides:

equals in class Object

toString

             public String toString()
            

Overrides:

toString in class Object