Windows环境搭建CenterNet

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

  1. 编译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"]

  1. 编译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
  1. 验证
    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底下

Add a Comment

电子邮件地址不会被公开。 必填项已用*标注