MultiPoseDetectionModelBase

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

Base class for pose detection models for detecting multiple poses per image.

Constructors

common

fun MultiPoseDetectionModelBase(modelKindDescription: String? = null)

Functions

convert

common

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

Converts raw model output to the result.

detectPoses

common

fun detectPoses(image: I, confidence: Float = 0.1f): <ERROR CLASS>

Detects poses for the given image with the given confidence.

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

MultiPoseDetectionModel