基于轻量级OpenPose的人体骨架提取

发布时间 2023-08-10 15:46:41作者: tlott

1、项目源码及权重文件下载

项目源码:https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch/blob/master/demo.py
权重文件:https://download.01.org/opencv/openvino_training_extensions/models/human_pose_estimation/checkpoint_iter_370000.pth
需要注意的是源码中的demo.py是应用时所需要的代码,如果需要训练才需要用到其余的文件。
image

2、demo.py运行

如果直接运行代码的话会遇到这种情况,如图1。
image
这是因为运行时需要传入对应参数,见代码底部。
image
此时可以选择pycharm右上角运行按钮处的配置按钮,如下图。
image
打开界面后,如图对Parameters进行参数输入。
image
部分参数输入如下:

--checkpoint-path checkpoint_iter_370000.pth --video 0  //打开摄像头,对拍摄物体进行骨架提取
--checkpoint-path checkpoint_iter_370000.pth --video 1.mp4  //选择视频文件进行骨架提取
--checkpoint-path checkpoint_iter_370000.pth --images 1.jpg  //选择图片进行骨架提取

当然也可以通过命令行进行调用

python demo.py --checkpoint-path checkpoint_iter_370000.pth --video 0
python demo.py --checkpoint-path checkpoint_iter_370000.pth --video 1.mp4
python demo.py --checkpoint-path checkpoint_iter_370000.pth --images 1.jpg

3、总结

使用已有权重文件的模型确实很简单,如果熟练的话可以在一个小时内完成本地部署,难的还是网络模型的搭建以及训练,这方面还需要加强