提交 e7e31d3e 编写于 作者: W wuzewu

Fix serving bug in human_pose_estimation_resnet50_mpii

上级 3e0717a1
......@@ -24,7 +24,7 @@ from human_pose_estimation_resnet50_mpii.pose_resnet import ResNet
author_email="paddle-dev@baidu.comi",
summary=
"Paddle implementation for the paper `Simple baselines for human pose estimation and tracking`, trained with the MPII dataset.",
version="1.1.0")
version="1.1.1")
class HumanPoseEstimation(hub.Module):
def _initialize(self):
self.default_pretrained_model_path = os.path.join(
......
......@@ -94,6 +94,7 @@ def postprocess(out_heatmaps, org_im, org_im_shape, org_im_path, output_dir,
print('image saved in {}'.format(save_im_name))
# articulation
preds = list(map(lambda pred: [int(_) for _ in pred], preds))
res['data']['left_ankle'] = list(preds[0])
res['data']['left_knee'] = list(preds[1])
res['data']['left_hip'] = list(preds[2])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册