说明: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中一直无法编译通过