centerne安装和实时人体姿态估计

环境

ubuntu18.04 TL
python1.6
pytorch 1.0.0
torchision0.2.2
ubuntu环境下我使用conda创建的虚拟环境,cudatoolkit
10.1.243(就是大家所说的CUDA)。cuda驱动是450版本,这个没有太大的影响只要他能支持起来cudatoolkit 10.1.243就可以。
以下是2条指令:

conda create -n  CenterNet  python=3.6
pip install torch==1.0.0  torchvision==0.2.2 

检测torch和torchvision是否安装成功

import torch
import torchvision as ts
print(torch.cuda.is_available())
print(ts.__version__)

使用pip3不是pip对以下两个模块进行安装,如果是虚拟环境需要下载alfred-py源代码python3 setup.py build develop进行安装,下载地址为alfred-py源代码下载,要安装到虚拟环境python下的set-packages包中,如我的路径为:/home/zhangzhenbo/anaconda3/envs/CenterNet/lib/python3.6/site-packages,在此路径下进行setup

sudo pip3 install alfred-py
sudo pip3 install absl-py

下载

centernet源代码下载地址,修改过的:
https://download.csdn.net/download/weixin_42912072/13109975
感谢:https://gitlab.com/StrangeAI 的贡献

DCNv2下载地址:
https://download.csdn.net/download/weixin_42912072/13109928
感谢:Forked from CharlesShang/DCNv2 的贡献

权重文件地址:
multi_pose_dla_3x.pth

编译和运行

此处主要有几个问题
1、编译
先对DCNv2进行setup编译,然后再在centernet中去./make.sh,make.sh需要授权,授权指令为chmod 777 make.sh
2、修改pose_dla_dcn.py和resnet_dcn.py中的import DCN的路径,可以在安装完毕后根据报错信息定位到行,修改成对应的路径即可,报错信息一般为no module named torch或者找不到**模块。
3 run
修改两处代码,否则默认为谷歌云的模型地址,而且会让你下载不成功,参考连接如下
https://blog.csdn.net/qq_41084756/article/details/101711202

python3 demo.py

测试没有报错即可进行下一步的摄像头实时测试

python demo.py multi_pose --demo webcam --load_model weights/multi_pose_dla_3x.pth

欢迎对此感兴趣的小伙伴转载和使用,时刻希望跟各位进行交流互动,可以流言联系我,有问题我会尽量提供帮助。

转载和引用:
https://github.com/xingyizhou/CenterNet
https://gitlab.com/StrangeAI/centernet
https://blog.csdn.net/qq_41084756/article/details/101711202
https://github.com/jinfagang/alfred
https://github.com/CharlesShang/DCNv2

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值