当前位置:网站首页>win10配置CenterNet环境
win10配置CenterNet环境
2022-08-09 18:44:00 【anthony-36】
最近在复现CenterNet时,碰到很多问题,花费了两天多的时间,但最后都一一解决了
一、准备
1.环境
Window10+VS2019+CUDA11.1+Python3.8+Pytorch1.8.0+Anaconda3+Pycharm
显卡:GTX2060;
2.论文
论文原文:https://arxiv.org/pdf/1904.07850.pdf(Objects as Points)
代码:https://github.com/xingyizhou/CenterNet
二、配置环境
下面所有的命令行都是在Pcharm中的Terminal中进行,主要执行的文件的环境就好
1.配置anaconda3+python3.8
conda create -n CenterNet python=3.6 #
activate CenterNet # 切换环境至CenterNet
这里安装好anaconda3+python3.8后,主要查看是否配置好环境变量。
2.安装CUDA和CUDANN
在选择CUDA时一定要考虑好Pytorch、CUDA、驱动和显卡之间的版本关系,
3.安装Pytorch
在CenterNet环境中
conda install pytorch==1.8.0 torchvision==0.9.0
安装好后,运行下列代码,查看安装是否完成
import torch
print(torch.version.cuda)
print(torch.__version__)
print(torch.cuda.is_available()) #查看cuda是否可用
print(torch.cuda.device_count()) #查看可行的cuda数目
4.安装CenterNet必备库
在CenterNet环境中
pip install requirements.txt
5.安装Visual Studio 2019
这个教程只看安装部分就好,安装好要查看是否配置好环境变量
三、编译NMS
cd CenterNet\src\lib\external
python setup.py build_ext --inplace
这里有问题出现:在编译NMS时,**cl: 命令行error D8021 :无效的数值参数“/Wno-cpp”请看踩坑2
四、编译DCNv2
先在文件夹中删除自带的DCNv2,在下面这个文件夹下,直接删除就好
CenterNet\src\lib\models\network
然后在Pycharm中,
cd CenterNet\src\lib\models\networks
git clone https://github.com/lbin/DCNv2/tree/pytorch_1.9
cd DCNv2
python setup.py build develop
五、运行dome.py
cd CenterNet/src
python demo.py ctdet --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth --debug 2
python demo.py multi_pose --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/multi_pose_dla_3x.pth --debug 2
运行过程中会在线下载ctdet_coco_dla_2x.pth,multi_pose_dla_3x.pth文件,可能会出现HTTP request请求超时报错。这里的两个文件,第一个是物体检测的,第二个是人体姿态检测的。
解决办法:事先下载好以上两个文件,放入项目models文件夹下.
若已经将文件下载至本地后,运行demo.py还是会在线下载模型,注释pose_dla_dcn文件中的第313,314行代码:
到这里就运行成功了。
六、踩坑
1.在编译DCNv2时,RuntimeError: Error compiling objects for extension
python setup.py build develop
出现:RuntimeError: Error compiling objects for extension
把DCNv2文件换为:
https://github.com/lbin/DCNv2/tree/pytorch_1.9
里面的pytorch可以在master中更改版本,我用的是pytorch1.8,这里pytorch1.9也可以用。
2.在编译NMS时,cl: 命令行error D8021 :无效的数值参数“/Wno-cpp”
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YjqyJAyU-1660014880101)(C:/Users/13611/AppData/Roaming/Typora/typora-user-images/image-20220808223544131.png)]
这一行代码注释掉
3.在编译DCNv2时,出现lc.exe有关问题
2.在编译NMS时,cl: 命令行error D8021 :无效的数值参数“/Wno-cpp”
[外链图片转存中…(img-YjqyJAyU-1660014880101)]
这一行代码注释掉
3.在编译DCNv2时,出现lc.exe有关问题
这里请查看VS2019的环境变量是否配置好,重启一下。
边栏推荐
- Why is the data of maxcompute garbled when imported into mysql?The table of mysql is the encoding of udf8mb4
- pat链表专题训练+搜索专题
- laravel 时区问题timezone
- 这年头还不来尝试线稿图视频??
- hdu 2094 产生冠军(STL map || 拓扑 || STL set)
- 使用Mock技术模拟数据
- 技术分享 | 接口自动化测试如何处理 Header cookie
- 明明加了唯一索引,为什么还是产生重复数据?
- 优秀的 Verilog/FPGA开源项目介绍(三十一)- OFDM
- Leetcode 739.每日温度 单调栈
猜你喜欢
对应运放 RC 滤波负反馈的波形
ebook下载 | 《 企业高管IT战略指南——企业为何要落地DevOps》
[免费专栏] Android安全之Xposed插件开发【从零手把手带】教程
[Free Column] Android Fragment Injection for Android Security
[免费专栏] Android安全之APK动态方式逆向应用【三种Smali注入方法】
Open Source Summer | List Details Display Based on Ruoyi Architecture
【Unity3D】2D动画
使用Mock技术模拟数据
Haven't tried line art videos this year??
重磅!上海985教授当选!全球仅4人!
随机推荐
[免费专栏] Android安全之Android Fragment注入
Codesys结构变量编程应用(STRUCT类型)
Why is the data of maxcompute garbled when imported into mysql?The table of mysql is the encoding of udf8mb4
嵌入式开发:使用FILL提高代码完整性
Swift--多条件排序
2022深圳(软考中级)系统集成项目管理工程师报名
[Free column] Xposed plug-in development for Android security [from scratch] tutorial
Office 365 Group概述以及创建方法
C#/VB.NET:从PowerPoint文档中提取文本和图片
Swift -- 数组高阶函数
队列题目:用队列实现栈
2021 RoboCom 世界机器人开发者大赛-本科组(决赛)
最新BEV感知基线 | 你确定需要激光雷达?(卡内基梅隆大学)
IDEA工具常用配置
日期及时间处理包 Carbon 在 Laravel 中的简单使用[通俗易懂]
2022.08.06_每日一题
Mysql 表结构变更方案对比及分析
DP-Differential Privacy概念介绍
数学建模——模拟退火
2022深圳(软考高级)信息系统项目管理师认证报名