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

working rotation hands

Browse Source
This commit is contained in:
Аркадий Стрельников
2021-12-23 15:15:45 +07:00
parent 47d9d7514a
commit 7d0a41dedd
3 changed files with 466 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
hpe_mp_class.py
View File

@@ -172,42 +172,48 @@ 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
#
# 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
if old_world:
rhandconvert(rhandlms, data, self.coef)
else:
rhandconvertwithrot(rhandlms, data, self.coef)
else:
rhandconverttranslation(data)
rhandrot(rhandlms, data, self.coef)
# if old_world:
# rhandconvert(rhandlms, data, self.coef)
# else:
# rhandconvert(rhandlms, data, self.coef)
# # rhandconverttranslation(data)
# # else:
# # rhandconverttranslation(data)
lhandlms = {}
if self.results_hol.left_hand_landmarks:
for id, lm in enumerate(self.results_hol.left_hand_landmarks.landmark):
lhandlms[id] = lm
if old_world:
lhandconvert(lhandlms, data, self.coef)
else:
lhandconvertwithrot(lhandlms, data, self.coef)
else:
lhandconverttranslation(data)
lhandrot(lhandlms,data,self.coef)
# if old_world:
# lhandconvert(lhandlms, data, self.coef)
# else:
# lhandconvert(lhandlms, data, self.coef)
# lhandconverttranslation(data)
# else:
# lhandconverttranslation(data)
else:
bodyaposelocal(data)
if self.holistic_use: