SinglePoseDetectionModelBase

abstract class SinglePoseDetectionModelBase<I>(modelKindDescription: String?) : OnnxHighLevelModel<I, <ERROR CLASS>>

Base class for pose detection models for detecting a single pose per image.

Constructors

common

fun SinglePoseDetectionModelBase(modelKindDescription: String? = null)

Functions

convert

common

open override fun convert(output: <ERROR CLASS>): <ERROR CLASS>

Converts raw model output to the result.

detectPose

common

fun detectPose(image: I): <ERROR CLASS>

Detects a pose for the given image.

initializeWith

common

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

common

open fun predict(input: I): <ERROR CLASS>

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

open override val modelKindDescription: String? = null

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

SinglePoseDetectionModel

Extensions

detectPose

android

fun SinglePoseDetectionModelBase<Bitmap>.detectPose(imageProxy: ImageProxy): DetectedPose

Detects a pose for the given imageProxy. Internal preprocessing is updated to rotate image to match target orientation. After prediction, internal preprocessing is restored to the original state.