基于 paddlehub 的模型姿态检测 human_pose_estimation_resnet50_mpii, 是一个人体关键点检测模型。将这个模型下载到本地,或者可以在百度飞桨上运行
import cv2
import paddlehub as hub
pose_estimation = hub.Module(name="human_pose_estimation_resnet50_mpii")
result= pose_estimation.keypoint_detection(images=[cv2.imread('C:/Users/86178/Desktop/2.jpg')], visualization=True)
print(result)
我们可以看到,这个模型将人定为 16 个关键点,模型识别效果还是不错的。