正文
win10 + cuda10.0 + pytorch1.2 + CenterNet 环境搭建
提示：扫一扫查出行【扫一扫了解最新限行尾号】
复制提示
心血来潮，想跑个 CenterNet 检测瞅瞅。。。麻蛋，有非官方层
一。下载好 CenterNet 源码https://github.com/xingyizhou/CenterNet
二。注意你需要 VS 2017，不然你后面不好玩三。build nmscd src\lib\externalpython setup.py build_ext --inplace
https://github.com/xingyizhou/CenterNet
三。build nmscd src\lib\externalpython setup.py build_ext --inplace
cd src\lib\externalpython 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，速度很慢，建议手动下载。
美滋滋，
