ONNXModelPreprocessor

class ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
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(input: Pair<FloatArray, TensorShape>): Pair<FloatArray, TensorShape>
Content copied to clipboard
getOutputShape
Link copied to clipboard
open override fun getOutputShape(inputShape: TensorShape): TensorShape
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.