CenterNet环境搭建与demo测试（包括ubuntu16.04版本和win10版本）

 

CenterNet官方github:https://github.com/xingyizhou/CenterNet

一、ubuntu16.04环境下的环境搭建与测试

环境：ubuntu16.04、pytorch0.4.0、cuda9.0

cuda配置：

参考我之前的博客：https://blog.csdn.net/weixin_40369473/article/details/82223775，其中还包括cuda版本共存与切换问题

pytorch安装：

conda install pytorch=0.4.0 cuda90 -c pytorch。（其他版本及安装方式见官网：https://pytorch.org/）

CenterNet配置：

参考官方github：https://github.com/xingyizhou/CenterNet/blob/master/readme/INSTALL.md

在下载好centernet安装包并安装好相应需要的包以后，要想运行demo，需要首先进行DCNv2（可变形卷积）和NMS的编译