标签：py -- demo DCNv2 配置 CenterNet 编译 Win10
目录
一、准备
1. 论文
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
- clone项目
git clone https://github.com/CharlesShang/DCNv2
- 原项目里的DCNv2是Pytorch 0.4.0的，所以需要将这个新下载的DCNv2替换原来的
C:\CenterNet\src\lib\models\networks\DCNv2文件夹 - 编译DCNv2
cd C:\CenterNet\src\lib\models\networks\DCNv2
python setup.py build develop
编译成功：
4. 编译NMS
- 进入到
cd C:\CenterNet\src\lib\external
- 执行以下代码，编译NMS
python setup.py build_ext --inplace
- 注意：
若出现错误：cl: 命令行 error D8021 :无效的数值参数“/Wno-cpp”
注释掉C:\CenterNet\src\lib\external中setup.pyline10
# extra_compile_args=["-Wno-cpp", "-Wno-unused-function"]
- 再次编译，成功！
三、运行demo
1. 下载模型
这里一共需要下载两个模型，都需要翻墙：
ctdet_coco_dla_2x.pth放到C:\CenterNet\models\
dla34-ba72cf86.pth是运行demo.py时自动下载的，可以手动下载后放到C:\Users\<USER_NAME>\.cache\torch\checkpoints\中
百度云链接：
https://pan.baidu.com/s/1gAyIVHL6U–IrIZolwst6A
提取码：lqo1
2. 运行demo.py
- 进入到
cd C:\CenterNet\src\
- 运行
python demo.py ctdet --demo ../images/ --load_model ../models/ctdet_coco_dla_2x.pth
出现小可爱！
- 使用
--debug 2显示heat map
python demo.py ctdet --demo ../images/ --load_model ../models/ctdet_coco_dla_2x.pth --debug 2
完结撒花！！！
四、坑
1. 编译DCNv2出错
UserWarning: Error checking compiler version for cl: [WinError 2]
- 更改
C:\Anaconda\Install\envs\centernet_pt15\Lib\site-packages\torch\utils\中cpp_extension.py第233行，加’ gbk’（gbk前有空格）
- 若依然报错，建议用Pytorch 1.4.0，我用Pytorch 1.5.0 和 1.5.1编译都会出错
Error checking compiler version for cl
2. 编译NMS出错
cl: 命令行 error D8021 :无效的数值参数“/Wno-cpp”
注释掉C:\CenterNet\src\lib\external中setup.pyline10
# extra_compile_args=["-Wno-cpp", "-Wno-unused-function"]
标签：py,--,demo,DCNv2,配置,CenterNet,编译,Win10 来源： https://blog.csdn.net/startllee/article/details/118655091
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。