sa/HPE_mediapipe
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Euler angles for Body

Browse Source
This commit is contained in:
Аркадий Стрельников
2022-01-10 14:08:05 +07:00
parent 7d0a41dedd
commit c6c2334d05
4 changed files with 2290 additions and 1909 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
hpe_mp_class.py
View File

@@ -13,6 +13,10 @@ logging.basicConfig(filename="hpe_mp_class.log", level=logging.INFO, format=FORM
logger = logging.getLogger("hpe_mp_class_logger")
from ModelUE4 import *
from ModelUE4_apose import bodyaposelocal
from ModelUE4_apose import bodyaposeworld
from ModelUE4_old import bodyconvert
from ModelUE4_old import bodyconvertlocal
class hpe_mp_class():
@@ -172,25 +176,27 @@ class hpe_mp_class():
if world:
bodyaposeworld(data)
if self.holistic_use:
# poseslms = {}
# maxy = 0
# if self.results_hol.pose_landmarks:
# for id, lm in enumerate(self.results_hol.pose_landmarks.landmark):
# poseslms[id] = lm
# if lm.y > maxy:
# maxy = lm.y
#
# if old_world:
# bodyconvert(poseslms, data, self.coef, maxy)
# else:
# bodyconvertwithrot(poseslms, data, self.coef, maxy)
#
poseslms = {}
maxy = 0
if self.results_hol.pose_landmarks:
for id, lm in enumerate(self.results_hol.pose_landmarks.landmark):
poseslms[id] = lm
if lm.y > maxy:
maxy = lm.y
bodyeuler(poseslms, data, self.coef)
# if old_world:
# bodyconvert(poseslms, data, self.coef, maxy)
# else:
# bodyconvertwithrot(poseslms, data, self.coef, maxy)
rhandlms = {}
if self.results_hol.right_hand_landmarks:
for id, lm in enumerate(self.results_hol.right_hand_landmarks.landmark):
rhandlms[id] = lm
rhandrot(rhandlms, data, self.coef)
rhandeuler(rhandlms, data, self.coef)
# if old_world:
# rhandconvert(rhandlms, data, self.coef)
@@ -205,7 +211,7 @@ class hpe_mp_class():
for id, lm in enumerate(self.results_hol.left_hand_landmarks.landmark):
lhandlms[id] = lm
lhandrot(lhandlms,data,self.coef)
lhandeuler(lhandlms,data,self.coef)
# if old_world:
# lhandconvert(lhandlms, data, self.coef)