Обработка результатов оценки позы Yolo в Android Tensorflow Lite API ⇐ Android

[Anonymous]

Я пытаюсь реализовать модель yolo11n-pose_float16.tflite в проекте Android Kotlin. Но я не могу получить правильные ключевые точки на выходе, потому что я что-то сделал неправильно. Я хочу нарисовать на экране только скелет, содержащий ключевые точки и соединения частей тела. Как мне извлечь выходной массив в мой объект Person?
data class Person(
var keyPoints: MutableList,
val score: Float
)

enum class BodyPart {
NOSE, LEFT_EYE, RIGHT_EYE, LEFT_EAR, RIGHT_EAR,
LEFT_SHOULDER, RIGHT_SHOULDER, LEFT_ELBOW, RIGHT_ELBOW,
LEFT_WRIST, RIGHT_WRIST, LEFT_HIP, RIGHT_HIP,
LEFT_KNEE, RIGHT_KNEE, LEFT_ANKLE, RIGHT_ANKLE
}

data class KeyPoint(val bodyPart: BodyPart, var coordinate: PointF, val score: Float)

// Here is my function:

private fun processOutput(output: FloatArray): List {

val persons = mutableListOf()
// How should I process output???
// output shape is [1,56,8400]

}


Подробнее здесь: https://stackoverflow.com/questions/790 ... w-lite-api