Certernet实现人体姿态估计(包括人体检测)的步骤记录

首先展示效果(此为官方图):
在这里插入图片描述
github链接:https://github.com/xingyizhou/CenterNet

1. 编译nms

Note:
若未编译nms,则运行demo时会报错ModuleNotFoundError: No module named ‘external.nms’

git clone https://github.com/xingyizhou/CenterNet.git
pip install -r requirements.txt  # 安装必要模块,可使用conda创建一个新的环境,需要torch 
cd $Centernet_HOME # (即centernet根目录)
cd src/lib/external
make

2. 编译DCNv2

Note:
源代码里给的是pytorch0.4版本的,如果用的是torch<=0.4版本可直接使用,若是pytorch1.x版本则需要下载DCNv2源码(链接地址)替换掉src/lib/model/networks里的DCNv2目录后再进行编译
使用torch1.x版本又使用此项目里自带的DCNv2,编译DCNv2后运行demo会报错ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.
我用的是pytorch1.1版本的

cd src/lib/model/networks
rm -rf DCNv2 # 移除旧版本DCNv2
git clone https://github.com/CharlesShang/DCNv2.git # 下载新版本DCNv2
cd DCNv2
./make.sh

Note:
以上编译在Linux上建议使用gcc4.9以上版本进行编译(尝试使用gcc4.8.5编译后运行demo检测视频时报错Segmentations fault (core dumped)

3. 模型

下载人体姿态估计模型multi_pose_dla_3x.pth 79M(链接:https://pan.baidu.com/s/1VUcK0CKZYANFnjn70WkTSw 密码:h1ls
运行demo时所需要的预训练模型dla34-ba72cf86.pth 60M(链接:https://pan.baidu.com/s/1A_zfUeQItuuCdNkzlpzZwg 密码:p4ye
,根据提示拷贝到.torch下的指定目录里

4. 运行demo

python demo.py multi_pose --demo 3.mp4 --load_model multi_pose_dla_3x.pth

5. c++调用(使用tensortRT加速)

后续更新转换模型供c++调用

  • 2
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值