win10 + cuda10.0 + pytorch1.2 + CenterNet 环境搭建

心血来潮，想跑个 CenterNet 检测瞅瞅。。。麻蛋，有非官方层

一。下载好 CenterNet 源码

https://github.com/xingyizhou/CenterNet

二。注意你需要 VS 2017，不然你后面不好玩

三。build nms

cd src\lib\external

python setup.py build_ext --inplace

额，你会报错滴。打开 setup.py， 注释下面这句话：

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

四。build DCN2

 原版的，玩不了了，因为是 pytorch 0.4 的。去这里下载覆盖下：

https://github.com/CharlesShang/DCNv2

然后，还没完哦，必须要修改 dcn_v2_cuda.cu 里一句代码：

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

好了，可以编译了：

cd src\lib\models\networks\DCNv2
python setup.py build develop

 好啦好啦，你可以跑 demo.py 啦。

五。下载模型

 这里有作者提供的很多模型，随便下载几个

https://github.com/xingyizhou/CenterNet/blob/master/readme/MODEL_ZOO.md

运行：

python demo.py ctdet --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth

这里会下载个 dla34-ba72cf86.pth，速度很慢，建议手动下载。

 

 美滋滋，

转载于:https://www.cnblogs.com/xuanyuyt/p/11430374.html