Github上搜了一下，發(fā)現CenterNet關注量最多的就是下面的實現，其實就是原作者的實現
https://github.com/xingyizhou/CenterNet
建議使用linux環(huán)境搭建，在這篇文章講述windows環(huán)境搭建時遇到的問題。
文章原文
我使用的環(huán)境是Win10 + Pytorch 1.2.0 + torchvision 0.4.0 + Cuda 10.1
關于Cuda的安裝可以看我的另一篇博文
步驟大致如下：
- 克隆CenterNet項目源碼
- 編譯 nms 和 DCNv2
- 下載預訓練模型
- 測試Demo
下面直接講項目編譯的過程
編譯與運行demo
- 編譯nms
cd CenterNet\src\lib\external
#python setup.py install
python setup.py build_ext --inplace
如果編譯有問題的話，需要注釋setup.py里面的這句代碼
#extra_compile_args=["-Wno-cpp", "-Wno-unused-function"]
- 編譯DCN2項目‘
先刪除自帶的DCNv2，重新clone一份新的
cd CenterNet\src\lib\models\networks
rm -rf DCNv2
git clone https://github.com/CharlesShang/DCNv2
cd DCNv2
修改下面代碼
vim cuda/dcn_v2_cuda.cu
# extern THCState *state;
THCState *state = at::globalContext().lazyInitCUDA();
開始編譯
python setup.py build develop
- 驗證
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
image
運行過程會提示下載文件失敗，可以手動粘貼地址，用瀏覽器下載。下載完畢放置到指定目錄即可。
下面記錄其他問題
安裝pytorch 需要訪問官網，選擇相應的電腦系統、cuda版本等信息生成安裝命令
官網地址如下：https://pytorch.org/get-started/locally/
按我的環(huán)境生成的命令如下：
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
python demo.py ctdet --demo /data/projects/CenterNet/images/18124840932_e42b3e377c_k.jpg --load_model ../models/ctdet_pascal_dla_384.pth
這里可以下載預先訓練的模型，用于測試。由于文件放在GoogleDrive，若有需要文件請聯系killfunst
https://github.com/xingyizhou/CenterNet/blob/master/readme/MODEL_ZOO.md
下載好的模型文件放置在項目models底下
