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

Comments and readme update

Browse Source
This commit is contained in:
Аркадий Стрельников
2022-02-08 20:02:14 +07:00
parent c6c2334d05
commit f73685fa55
6 changed files with 141 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
hpe_mp_class.py
View File

@@ -15,8 +15,8 @@ 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
#from ModelUE4_old import bodyconvert
#from ModelUE4_old import bodyconvertlocal
class hpe_mp_class():
@@ -169,13 +169,14 @@ class hpe_mp_class():
# try:
if apose:
if world:
bodyaposeworld(data)
bodyaposeworld(data) # APose world
else:
bodyaposelocal(data)
bodyaposelocal(data) # APose local
else:
if world:
bodyaposeworld(data)
if self.holistic_use:
# body converting
poseslms = {}
maxy = 0
if self.results_hol.pose_landmarks:
@@ -184,6 +185,7 @@ class hpe_mp_class():
if lm.y > maxy:
maxy = lm.y
# only rotation method
bodyeuler(poseslms, data, self.coef)
# if old_world:
@@ -191,11 +193,13 @@ class hpe_mp_class():
# else:
# bodyconvertwithrot(poseslms, data, self.coef, maxy)
# right hand converting
rhandlms = {}
if self.results_hol.right_hand_landmarks:
for id, lm in enumerate(self.results_hol.right_hand_landmarks.landmark):
rhandlms[id] = lm
# only rotation method
rhandeuler(rhandlms, data, self.coef)
# if old_world:
@@ -206,11 +210,13 @@ class hpe_mp_class():
# # else:
# # rhandconverttranslation(data)
# left hand converting
lhandlms = {}
if self.results_hol.left_hand_landmarks:
for id, lm in enumerate(self.results_hol.left_hand_landmarks.landmark):
lhandlms[id] = lm
# only rotation method
lhandeuler(lhandlms,data,self.coef)
# if old_world:
@@ -222,18 +228,18 @@ class hpe_mp_class():
# lhandconverttranslation(data)
else:
bodyaposelocal(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
bodyconvert(poseslms, data, self.coef, maxy)
bodyconvertlocal(poseslms, data, self.coef, maxy)
# 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
#
# bodyconvert(poseslms, data, self.coef, maxy)
#
# bodyconvertlocal(poseslms, data, self.coef, maxy)
# except Exception as err:
# logger.exception("Error json converting hpe class: " + str(err))