Package com.itextpdf.pdfocr.onnx.orientation

Class OnnxOrientationPredictorProperties

java.lang.Object

com.itextpdf.pdfocr.onnx.AbstractOnnxPredictorProperties

com.itextpdf.pdfocr.onnx.orientation.OnnxOrientationPredictorProperties

       public class OnnxOrientationPredictorProperties extends AbstractOnnxPredictorProperties
      

Properties for configuring crop orientation ONNX models.

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

Field Summary

Fields inherited from class com.itextpdf.pdfocr.onnx.AbstractOnnxPredictorProperties

DEFAULT_ORT_SESSION_CREATOR, inputProperties, modelPath, ortSessionOptionsCreator

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
boolean	equals(Object o)
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.
static OnnxOrientationPredictorProperties	mobileNetV3(String modelPath, IOrtSessionOptionsCreator ortSessionOptionsCreator)	Creates a new crop orientation properties object for existing pre-trained MobileNetV3 models, stored on disk.
String	toString()

Methods inherited from class com.itextpdf.pdfocr.onnx.AbstractOnnxPredictorProperties

getInputProperties, getModelPath, getOrtSessionOptionsCreator

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

OnnxOrientationPredictorProperties

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

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

ortSessionOptionsCreator - ONNX runtime session options creator

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

mobileNetV3

             public static OnnxOrientationPredictorProperties mobileNetV3
             (String modelPath, IOrtSessionOptionsCreator ortSessionOptionsCreator)
            

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

ortSessionOptionsCreator - the ONNX runtime session options creator

Returns:

a new crop orientation properties object for a MobileNetV3 model

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