MultiPoseDetectionModel

class MultiPoseDetectionModel : OnnxInferenceModel

MultiPoseDetectionModel is an ultra-fast and accurate model that detects 6 persons with 17 keypoints and 18 basic edges of a body for each of them.

It internally uses ONNXModels.PoseDetection.MoveNetMultiPoseLighting under the hood to make predictions.

Constructors

fun MultiPoseDetectionModel()

Functions

open override fun close()

Releases the ONNXRuntime - related resources.

copy

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

detectPoses

fun detectPoses(imageFile: File, confidence: Float = 0.1f): MultiPoseDetectionResult

fun detectPoses(inputData: FloatArray, confidence: Float = 0.005f): MultiPoseDetectionResult

evaluate

fun evaluate(dataset: Dataset, metric: Metrics): Double

predict

open override fun predict(inputData: FloatArray): Int

fun predict(dataset: Dataset): List<Int>

fun predict(inputData: FloatArray, inputTensorName: String, outputTensorName: String): Int

Predicts the class of inputData.

predictRaw

fun predictRaw(inputData: FloatArray): Map<String, Any>

Returns list of multidimensional arrays with data from model outputs.

predictRawWithShapes

fun predictRawWithShapes(inputData: FloatArray): List<Pair<FloatBuffer, LongArray>>

Returns list of pairs from model outputs.

predictSoftly

fun predictSoftly(inputData: FloatArray): FloatArray

Predicts vector of probabilities instead of specific class in predict method.

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

reshape

open override fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

toString

open override fun toString(): String

Properties

inputDataType

lateinit var inputDataType: OnnxJavaType

Data type for input tensor.

inputDimensions

open override val inputDimensions: LongArray

inputShape

lateinit var inputShape: LongArray

Data shape for prediction.

name

var name: String?

outputDataType

lateinit var outputDataType: OnnxJavaType

Data type for output tensor.

outputShape

lateinit var outputShape: LongArray

Data shape for prediction.