【centernet】编译centernet支持pytorch1.2.0

说明：centernet项目是在pytorch0.4.1上测试的，这里将其在pytorch1.2.0上编译。主要的问题是DCNv2拓展。

转载自： https://github.com/xingyizhou/CenterNet/issues/7

1. build nms

cd CenterNet\src\lib\external
#python setup.py install
python setup.py build_ext --inplace

2. clone and build original DCN2

cd CenterNet\src\lib\models\networks
rm -rf DCNv2
git clone https://github.com/CharlesShang/DCNv2
cd DCNv2

python setup.py build develop

说明:

• 这里不用改dcn_v2_cuda.cu，因为DCNv2项目已经更新进去了

可能问题
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

• 这是因为gcc版本的问题导致的，错误时是用6.3编译的。将gcc切换回5.4就OK了。

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

4. 问题

• 不知道为什么在pytorch1.1.0中一直无法编译通过