基于pytorch-openpose框架的的人体姿态检测

1.在github上下载pytorch-openpose压缩包,网址为:
https://github.com/Hzzone/pytorch-openpose
在这里插入图片描述

2.解压后所需要cuda的环境:cuda 10.0,cudnn 10.0,NVIDIA显卡驱动程序安装(自行下载,如需要我给可在下方留言),全部安装完成后在命令行里面输入nvcc -V,查看是否安装好所需要的cuda版本,如图。
全部装好后的cuda版本显示
3.在原作者里面使用百度网盘下载作者给的预训练模型,模型下载地址,下载完成后,解压后将所有模型放入model文件夹,如图在这里插入图片描述
4.所需要的python是3.7的版本,所需要的Python库有:

pip install numpy
pip install matplotlib
pip install opencv-python
pip install scipy
pip install scikit-image
pip install tqdm

这些库没有什么具体的版本,直接通过该命令下载,现在下载torch和torchvision,使用该命令直接同时下载:

pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

最后需要需要下载cudatoolkit=10.0和cudnn=7.6.5的json文件,这两个文件很重要,是调用GPU的关键文件,使用下面命令可直接下载:

conda install cudatoolkit=10.0
conda install cudnn=7.6.5

在这里插入图片描述
5.如何运行,先下载一个视频,放入项目中,这里我下载了五个做测试用(小技巧:使用you-get库下载视频素材,不懂的可以去了解)在这里插入图片描述
6.打开demo_camera.py文件,在里面修改视频来源,将cap = cv2.VideoCapture(0)修改成cap = cv2.VideoCapture(“5.mp4”),如图在这里插入图片描述
7.直接运行demo_camera.py文件,运行成功,画面如图在这里插入图片描述
8.优点和缺点,优点:识别肢体的姿态准确率高达百分之90多,缺点:调用的GPU只占百分之零点一,速度很慢,需要去优化速度,再会。

9.成功了请求兄弟们一键三连,感谢。

评论 76
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

萧鼎

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值