win10+cuda10.0+pytorch1.4测试CenterNet

本文主要是介绍win10+cuda10.0+pytorch1.4测试CenterNet，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

整理好的代码和权重都在百度云链接，记得这个是被我加了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\externalpython 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，速度很慢，建议手动下载。下载不行的参考首部的百度云链接。

 

这篇关于win10+cuda10.0+pytorch1.4测试CenterNet的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---