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

接下去準備在這個模型基礎上搞些好玩的東西出來！
敬請期待！