Centernet在windows上的编译方法

环境: win10 vs2017 pytorch1.1 cuda10

完整工程centernet_window完整工程下载地址：CenterNet-windows

1、编译DCN库：

由于CenterNet使用了可变卷积网络，而在pytorch上并没有集成，所以使用前要先进行Dcn的编译。

a、准备跟pytorch1.0对应版本的DCNv2:

pytorch1.0现在已经不支持DCNv1的结构了，需要使用的是DCNv2的结构，下载地址：

DCNv2

b、修改src/cuda/下的dcn_v2_cuda.cu文件在#L11改为：

//extern THCState *state;
THCState *state = at::globalContext().lazyInitCUDA();

d、然后回退到set.py目录进行编译：

python setup.py build develop

    注意：至此，整个项目编译完成了。如果有出现一些包cuda相关的错误，有种情况：是cuda版本太旧，其中cuda版本一定要大于cuda9,不包括cuda9，因为cuda9对VS2017的支持不好，会报错。解决方法装多一个cuda版本即可。

2、编译nms库：

a、cd路径到external/目录下。

b、修改set.py里的代码，在#L10处改为：

#extra_compile_args=["-Wno-cpp", "-Wno-unused-function"]

c、使用下面命令进行编译：

python setup.py build_ext --inplace

  自此编译完成。

3、根据上面的编译结构可以进行test，查看是否成功：

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

其中的模型文件ctdet_coco_dla_2x.pth需要下载。结果如下：