ONNXModelPreprocessor

class ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int) : Preprocessor
Content copied to clipboard

Applies the given onnxModel as a preprocessing stage.

Constructors

ONNXModelPreprocessor
Link copied to clipboard
fun ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int = 0)
Content copied to clipboard

Functions

apply
Link copied to clipboard
open override fun apply(data: FloatArray, inputShape: ImageShape): FloatArray
Content copied to clipboard

Properties

onnxModel
Link copied to clipboard
var onnxModel: OnnxInferenceModel?
Content copied to clipboard

ONNX model. It could have multiple outputs.

outputIndex
Link copied to clipboard
var outputIndex: Int = 0
Content copied to clipboard

Index of the output to be passed forward.