Win10配置CenterNet

一、准备

1. 论文

• 论文题目：Object as Point.
• 代码：https://github.com/xingyizhou/CenterNet

2. 环境

• 系统：Win10
• 显卡：GTX 1080ti
• CUDA 10.1
• cuDNN 7.6.5
• Visual Studio 2017
• Pytorch 1.4.0
• Anaconda3

二、配置环境

1. clone项目

CenterNet:

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

2. 配置虚拟环境

• 创建虚拟环境

conda create -n centernet python=3.6

conda activate centernet

• 安装项目依赖包

cd ./CenterNet

pip install -r requirements.txt

• 安装Pytorch 1.4.0

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

3. 编译DCNv2

• clo