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,速度很慢,建議手動下載。

 

 美滋滋,


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM