环境
ubuntu18.04 TL
python1.6
pytorch 1.0.0
torchision0.2.2
ubuntu环境下我使用conda创建的虚拟环境,cudatoolkit10.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