Win10 成功搭建mmdetection2.0 以及测试

注：由于图片上传不了，各位看文字吧！！！

软件总览：

• CUDA:cuda_10.1.105_418.96_win10
• CUDNN:cudnn-10.1-windows10-x64-v8.0.3.33
• VS2017:vs_community__799919136.1602297293
• Anaconda:Anaconda3-5.3.0-Windows-x86_64
• Python:python-3.7.0-amd64
• pytorch=1.3
• torchvision=0.4
• 下载：cocoapi    mmcv-0.5.8    mmdetection2.0包文件

 

安装总览（具体安装细节可以自行百度）

 

步骤一：CUDA10.1下载与安装

https://developer.nvidia.com/cuda-10.1-download-archive-base

步骤二：CUDNN下载与安装

https://developer.nvidia.com/rdp/cudnn-archive

步骤三：VS2017下载与安装

https://docs.microsoft.com/en-us/visualstudio/productinfo/vs2017-system-requirements-vs

步骤四：Anaconda3下载与安装

https://repo.anaconda.com/archive/index.html

步骤五：Python3下载与安装

https://www.python.org/downloads/release/python-370/

步骤六：打开anaconda3的Anaconda Prompt界面

步骤七：依次执行输入（安装需要时间）

    conda create -n mmd2 python=3.7  创建mmdetection虚拟环境，名字自己随意

    conda activate mmd2   激活mmdetection虚拟环境

   conda install pytorch=1.3 torchvision cudatoolkit=10.1 -c pytorch   安装pytorch，必须对应与cudatoolkit版本

   pip install cython opencv-python pillow  matplotlib  cython版本不需要定，默认安装即可

 

 

步骤八：修改文件（替换虚拟环境mmd2中的文件可以私聊）

1、修改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())
2、D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\include\c10\util
     替换 flat_hash_map.h
     D:\ProgramData\Anaconda3\envs\mmd2\Lib\site-packages\torch\include\c10\util
     替换 order_preserving_flat_hash_map.h

步骤九：安装mmcv （cd进入文件所在目录下）

   cd mmcv-0.5.8

   pip install -e .

步骤十：安装pycocotools（cd进入文件所在目录下）
   cd cocoapi/PythonAPI
   python setup.py build_ext install

步骤十一：进入mmdetection2.0（cd进入文件所在目录下）
   pip install -r requirements.txt

步骤十二：修改mmdetection2.0文件中的setup.py文件
修改setup.py文件中CUDAExtension中extra_compile_args 相关代码，增加cxx的："-DMS_WIN64","-MD"

步骤十三：编译setup.py文件（需要等待几分钟）

 python setup.py develop

步骤十四：测试mmdetection2.0安装成功

（1）摄像头检测demo：python demo/webcam_demo.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth
（2）图像检测demo：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