未验证 提交 49379331 编写于 作者: Z zhiboniu 提交者: GitHub

update kpt and action data collector; (#5464)

test=document_fix
上级 abb60be9
...@@ -86,9 +86,10 @@ class DataCollector(object): ...@@ -86,9 +86,10 @@ class DataCollector(object):
if attr_res: if attr_res:
self.collector[ids]["attrs"].append(attr_res['output'][idx]) self.collector[ids]["attrs"].append(attr_res['output'][idx])
if kpt_res: 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: if action_res:
self.collector[ids]["actions"].append(action_res['output'][idx]) self.collector[ids]["actions"].append(action_res[idx + 1])
else: else:
# action model generate result per X frames, Not available every frames # action model generate result per X frames, Not available every frames
self.collector[ids]["actions"].append(None) self.collector[ids]["actions"].append(None)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册