Win10系统搭建mmdetection框架

背景:

做目标检测常用的框架可以mmdetection,但是官方的mmdetection框架是基于Ubuntu系统的,以下提供Window10系统安装mmdetection框架的解决方案。(首先感谢一下‘小鸡炖技术’B站UP主)


步骤:

【1】首先安装Visual Studio 2015: [安装地址](https://my.visualstudio.com/Downloads?q=visual%20studio%202015&wt.mc_id=o~msft~vscom~older-downloads)

在这里插入图片描述
下载Visual Studio Community 2015,这个版本是免费的,所以不涉及版权问题。
【2】安装Cuda 10.1,这个过程网络上挺多的,注意最后要把cuda环境配置到系统环境变量中,配置好之后,重启!重启!重启!
【3】下载mmdetection所需的安装包,下面提供网盘地址:
https://pan.baidu.com/s/1TP7MgFeTmUSYBsD99pY9Pw 提取码:hl0m
【4】创建一个环境:使用python3.7,我自己试的时候python3.6会报错。因此用下面命令创建一个环境:

conda create -n mmd2 python=3.7

【5】激活mmd2环境:

conda activate mmd2

【6】安装pytorch=1.3,因为系统装的是cuda10.1,因此安装pytorch=1.3,在cuda10.0下无法安装pytorch=1.3,装起来的版本后面会带有cu92,所以直接就上以下命令:

conda install pytorch=1.3 torchvision cudatoolkit=10.1 -c pytorch

【7】安装一些依赖的库:

pip install cython opencv-python pillow  matplotlib

【8】修改D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\utils\cpp_extension.py
185行:match = re.search(r’(\d+).(\d+).(\d+)’, compiler_info.decode(“utf8”,“ignore”).strip())
【9】用下载下来的网盘文件中,在mmdetection2.0文件夹下找到flat_hash_map.h以及order_preserving_flat_hash_map.h 然后进入刚才创建的环境目录,我的是这样的:D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\include\c10\util 然后用下载下来的两个文件,替换环境目录下的文件。(注意记得备份原文件,可以在出错的时候还原)
【10】用cd命令切换到mmcv-0.5.8文件夹下,执行下列命令:

pip install -e .    #后面有个点,别漏了

【11】切换到cd cocoapi/PythonAPI,这个是你下载下来文件的地址,cd到这个路径就可以了。

  python setup.py build_ext install

【12】cd 进入mmdetection2.0文件夹

pip install -r requirements.txt

【13】最后执行

python setup.py develop

【14】编译成功后会显示
在这里插入图片描述
【最后测试一下】

python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth 

在这里插入图片描述


原因分析:

严格按照以上配置是能够完成的,如果有其他问题,欢迎评论区留言。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值