Я пытаюсь реализовать модель 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
