关于在windows上完成目标检测模型Centernet 以及自己数据的训练

关于在windows上完成目标检测模型Centernet 以及自己数据的训练

配置设置

最近在windows上复现了centernet模型，跑通了demo，并对自己的数据进行了训练，这里对过程进行记录。
配置：Windows 10，VS2015，Anaconda3.5.1，python3.6，torch1.1，Cuda10.0，Cudnn7.5。（VS2015和python3.6应该是对应torch1.1）可以直接下载whl文件进行本地安装。文中涉及到的所有额外文件，包括安装包，py文件，模型文件，需要替换的文件，我都将打包上传。
模型：https://github.com/xingyizhou/CenterNet
将源码下载之后保存到英文目录下，因为代码中的cv2包的cv2.imread()只能从英文路径中读取文件，类似于：

模型复现/demo.py的运行

1

在命令行窗口中进入对应的源码文件下，对requirements.txt内的包进行安装：
pip install -r requirements.txt

2 编译DCNv2文件

（1）下载 https://github.com/CharlesShang/DCNv2 内的源码，然后将DCNv2文件夹替换CenterNet\src\lib\models\networks\DCNv2文件夹
（2）修改 DCN文件2\src\cuda\dcn_v2_cuda.cu：
//extern THCState *state; // 注释该行
THCState *state = at::globalContext