PaddleHub人体姿态检测模型pose_resnet50_mpii

• 2020 年 4 月 30 日
• AI
• 姿态检测

姿态检测还是挺有意思的，在 paddlehub 上有直接可以拿来用的模型：
pose_resnet50_mpii

随便网上找了张图片试了一下效果还行：


代码非常简单：

import paddlehub as hub
module = hub.Module(name="pose_resnet50_mpii")

# set input dict
input_dict = {"image": ['path/to/image']}

# execute predict and print the result
results = module.keypoint_detection(data=input_dict)
for result in results:
    print(result['path'])
    print(result['data'])

接下去准备在这个模型基础上搞些好玩的东西出来！
敬请期待！