心血來潮,想跑個 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,速度很慢,建議手動下載。
美滋滋,