ImageRecognitionModel

The light-weight API for Classification task with one of the Model Hub models.

Constructors

android

fun ImageRecognitionModel(internalModel: OnnxInferenceModel, channelsFirst: Boolean, preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>> = Identity(), modelKindDescription: String? = null, classLabels: Map<Int, String> = Imagenet.V1k.labels())

android

open override fun close()

copy

android

open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): InferenceModel

initializeWith

android

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

android

open override fun predict(inputData: FloatArray): Int

predictObject

android

predictSoftly

android

open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray

predictTopKObjects

android

reshape

android

open override fun reshape(vararg dims: Long)

summary

android

open override fun summary(): ModelSummary

inputDimensions

android

targetRotation

android

open override var targetRotation: Int = 0

Target image rotation.