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

Test angles before

Browse Source
This commit is contained in:
Arkadiy Strelnikov
2021-12-12 20:09:02 +07:00
parent a28855b99e
commit 7cc5954609
4 changed files with 34 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
hpe_mp_class.py
View File

@@ -160,7 +160,7 @@ class hpe_mp_class():
except Exception as err:
logger.exception("Error scaling hpe class: " + str(err))
def getJSON(self, apose=False, world=True):
def getJSON(self, apose=False, world=True, old_world=False):
data = {}
try:
if apose:
@@ -180,16 +180,20 @@ class hpe_mp_class():
if lm.y > maxy:
maxy = lm.y
# bodyconvert(poseslms, data, self.coef, maxy)
bodyconvertwithrot(poseslms, data, self.coef, maxy)
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
# rhandconvert(rhandlms, data, self.coef)
rhandconvertwithrot(rhandlms, data, self.coef)
if old_world:
rhandconvert(rhandlms, data, self.coef)
else:
rhandconvertwithrot(rhandlms, data, self.coef)
else:
rhandconverttranslation(data)
@@ -198,8 +202,10 @@ class hpe_mp_class():
for id, lm in enumerate(self.results_hol.left_hand_landmarks.landmark):
lhandlms[id] = lm
lhandconvert(lhandlms, data, self.coef)
lhandconvertwithrot(lhandlms, data, self.coef)
if old_world:
lhandconvert(lhandlms, data, self.coef)
else:
lhandconvertwithrot(lhandlms, data, self.coef)
else:
lhandconverttranslation(data)
else: