【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項目已經更新進去了

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中一直無法編譯通過