天天看點

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 

繼續閱讀