From 49379331ad6dbb732c2942492bd66f8bd92728fb Mon Sep 17 00:00:00 2001
From: zhiboniu <31800336+zhiboniu@users.noreply.github.com>
Date: Sat, 26 Mar 2022 14:39:06 +0800
Subject: [PATCH] update kpt and action data collector; (#5464)

test=document_fix
---
 deploy/pphuman/datacollector.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/deploy/pphuman/datacollector.py b/deploy/pphuman/datacollector.py
index 95e1ab575..cd459aad0 100644
--- a/deploy/pphuman/datacollector.py
+++ b/deploy/pphuman/datacollector.py
@@ -86,9 +86,10 @@ class DataCollector(object):
             if attr_res:
                 self.collector[ids]["attrs"].append(attr_res['output'][idx])
             if kpt_res:
-                self.collector[ids]["kpts"].append(kpt_res['keypoint'][idx])
+                self.collector[ids]["kpts"].append(
+                    [kpt_res['keypoint'][0][idx], kpt_res['keypoint'][1][idx]])
             if action_res:
-                self.collector[ids]["actions"].append(action_res['output'][idx])
+                self.collector[ids]["actions"].append(action_res[idx + 1])
             else:
                 # action model generate result per X frames, Not available every frames
                 self.collector[ids]["actions"].append(None)
-- 
GitLab