使用COCO数据集上的预训练模型进行姿势检测

时间:2020-05-04 18:44:20

标签: python deep-learning computer-vision pre-trained-model openpose

我目前正在使用Coco数据集中的Open Pose相关模型来预测我拥有的图像的关键点。我正在使用以下两个链接中的代码来尝试对我拥有的自定义数据进行姿态检测。

https://www.learnopencv.com/deep-learning-based-human-pose-estimation-using-opencv-cpp-python/ https://github.com/spmallick/learnopencv/blob/master/OpenPose/OpenPose_Notebook.ipynb

输入视频中,我有一个人在躺着锻炼。我在下面提供了一张图片:Input image

但是,如下图所示,检测到的关键点并不包括膝盖等人体中的几个点:

Output image

有人可以提供如何改进这一点的指示吗?还是有一个更好的数据集或另一个可以提供更好准确性的预训练模型?谢谢。

0 个答案:

没有答案