PaddleHub人体姿态检测模型pose_resnet50_mpii

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

随便网上找了张图片试了一下效果还行:
PaddleHub人体姿态检测模型pose_resnet50_mpii_第1张图片
PaddleHub人体姿态检测模型pose_resnet50_mpii_第2张图片
代码非常简单:

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'])

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

你可能感兴趣的:(计算机视觉,PaddlePaddle)