Как определить угол между двумя линиями в режиме реального времени по тензорному потоку

Question

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

Answer 1

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