Блог

Наш дипломный проект заключается в том, чтобы отслеживать движения человека в режиме реального времени во время выполнения упражнения, а затем давать ему обратную связь, если это упражнение правильное или неправильное, и инструкции, чтобы сделать его идеальным, поэтому мы разделили код на две части. Первый — это оценка всех ключевых точек пользователя (joint), которые выполняются в репозитории здесь с помощью tensorflow : https://github.com/ildoonet/tf-pose-estimation и мы запускаем этот код, вторая часть которого заключается в выполнении небольшого упражнения, такого как привязка руки (части кисти) на 90 градусов к плечу, поэтому нам нужно определить углы между конечностями и дать инструкции (голосовые или письменные команды) в соответствии с этим углом, чтобы кто-нибудь мог помочь с идеей или известным кодом?

В модуле tf_pose.estimator есть класс Human, список которого возвращается из метода PoseEstimator.inference(). Из Human.body_parts вы можете получить все позиции и соединения — joints. Затем вы можете вычислить углы между деталями в соединениях. Должна быть какая-то довольно простая тригонометрия с функцией math.atan2, чтобы получить абсолютный угол к положительной оси x. Затем вы можете вычислить, как расположены углы относительно друг друга в ссылке, например, начиная с плеча.