Github上搜了一下,发现CenterNet关注量最多的就是下面的实现,其实就是原作者的实现
https://github.com/xingyizhou/CenterNet
建议使用linux环境搭建,在这篇文章讲述windows环境搭建时遇到的问题。
文章原文
我使用的环境是Win10 + Pytorch 1.2.0 + torchvision 0.4.0 + Cuda 10.1
关于Cuda的安装可以看我的另一篇博文
步骤大致如下:
- 克隆CenterNet项目源码
- 编译 nms 和 DCNv2
- 下载预训练模型
- 测试Demo
下面直接讲项目编译的过程
编译与运行demo
- 编译nms
cd CenterNet\src\lib\external
#python setup.py install
python setup.py build_ext --inplace
如果编译有问题的话,需要注释setup.py里面的这句代码
#extra_compile_args=["-Wno-cpp", "-Wno-unused-function"]
- 编译DCN2项目‘
先删除自带的DCNv2,重新clone一份新的
cd CenterNet\src\lib\models\networks
rm -rf DCNv2
git clone https://github.com/CharlesShang/DCNv2
cd DCNv2
修改下面代码
vim cuda/dcn_v2_cuda.cu
# extern THCState *state;
THCState *state = at::globalContext().lazyInitCUDA();
开始编译
python setup.py build develop
- 验证
cd CenterNet/src
python demo.py ctdet --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth --debug 2
python demo.py multi_pose --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/multi_pose_dla_3x.pth --debug 2
image
运行过程会提示下载文件失败,可以手动粘贴地址,用浏览器下载。下载完毕放置到指定目录即可。
下面记录其他问题
安装pytorch 需要访问官网,选择相应的电脑系统、cuda版本等信息生成安装命令
官网地址如下:https://pytorch.org/get-started/locally/
按我的环境生成的命令如下:
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
python demo.py ctdet --demo /data/projects/CenterNet/images/18124840932_e42b3e377c_k.jpg --load_model …/models/ctdet_pascal_dla_384.pth
这里可以下载预先训练的模型,用于测试。由于文件放在GoogleDrive,若有需要文件请联系killfunst
https://github.com/xingyizhou/CenterNet/blob/master/readme/MODEL_ZOO.md
下载好的模型文件放置在项目models底下