注:由于圖檔上傳不了,各位看文字吧!!!
軟體總覽:
- 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