CenterNet 在win10+cuda10.0+pytorch1.1.0下搭建运行环境

1下载源码

https://github.com/xingyizhou/CenterNet.git

2下载DcnV2源码

https://github.com/CharlesShang/DCNv2.git

原版的版本不对  需要替换

3安装VS2015或2017

否则编译不了

4编译DCNV2

将/src/lib/models/networks/DCNv2中的文件清空

使用下载的DCNv2替换

修改/DCNv2/src/cuda/dcn_v2_cuda.cu文件，如下

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

然后

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

最后提示Finished…………则成功。

5编译NMS

位置在

cd src\lib\external

修改setup.py

删掉一行

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

然后运行

python setup.py build_ext --inplace

6.下载模型并运行

模型下载地址

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

下载后放在models文件夹下

运行src下的demo.py

python demo.py ctdet --demo  ../images/16004479832_a748d55f21_k.jpg  --load_model ../models/ctdet_coco_dla_2x.pth

第一次运行会下载一个权重，耐心等待

然后

来源：CSDN

作者：随便写点笔记

链接：https://blog.csdn.net/weixin_41977938/article/details/103460919