整理好的代码和权重都在百度云链接,记得这个是被我加了lib的路径的,直接运行demo.py即可。
链接:https://pan.baidu.com/s/18hBS21hF1znXJkDyneELTw
提取码:f4ws
一、下载好 CenterNet 源码
下载工程加速方式参考https://blog.csdn.net/u013066730/article/details/107164870
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 1.0(千万别下载错了) 的。去这里下载覆盖下:
https://github.com/CharlesShang/DCNv2/tree/pytorch_1.0
好了,可以编译了:
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,速度很慢,建议手动下载。下载不行的参考首部的百度云链接。