OnnxHighLevelModel

interface OnnxHighLevelModel<I, R> : ExecutionProviderCompatible

Base class for pre-trained high-level onnx models.

Parameters

input type

output type

Functions

convert

common

abstract fun convert(output: <ERROR CLASS>): R

Converts raw model output to the result.

initializeWith

common

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

common

open fun predict(input: I): R

Makes prediction on the given input.

summary

common

open fun summary(): <ERROR CLASS>

Properties

internalModel

common

abstract val internalModel: OnnxInferenceModel

Model used to make predictions.

modelKindDescription

common

abstract val modelKindDescription: String?

High-level description of the model. Used for model summary printing. For the model hub models it equals to the string representation of the OnnxModelType

preprocessing

common

abstract val preprocessing: <ERROR CLASS><I, <ERROR CLASS><FloatArray, <ERROR CLASS>>>

Preprocessing operation specific to this model.

Inheritors

FaceAlignmentModelBase

FaceDetectionModelBase

ObjectDetectionModelBase

MultiPoseDetectionModelBase

SinglePoseDetectionModelBase