mmdetection2.6的安裝
mmdetection的安裝
mmdetection的github鏈接：//github.com/open-mmlab/mmdetection
mmdetection的官方文檔：//mmdetection.readthedocs.io/en/latest/
1. mmdetection的安裝
- 創建conda環境
conda create -n mmdetection python=3.7 -y
source activate mmdetection
- 安裝pytorch及torchvision
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
#也可以上官網選擇特定的版本
# //pytorch.org/
- 安裝mmcv
可以在這個網址//github.com/open-mmlab/mmcv#install-with-pip, 查找自己的環境所需的mmcv包。
pip install mmcv-full==latest+torch1.6.0+cu102 -f //download.openmmlab.com/mmcv/dist/index.html
- clone mmdetection的程式碼庫
git clone //github.com/open-mmlab/mmdetection.git
cd mmdetection培養桃紅
- 安裝mmdetection
pip install -r requirements/build.txt
pip install -v -e . # 這裡有個點不能忘了
這裡採用pip install -v -e .安裝的是最小運行依賴， 如果想要使用可選擇的依賴，例如albumentations 和 imagecorruptions可以直接運行下邊的程式碼
pip install -v -e .[optional]
2. 驗證mmdetection是否安裝成功
在demo文件夾下創建demo.py,輸入以下的內容：
from mmdet.apis import init_detector, inference_detector
config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
device = 'cuda:0'
# init a detector
model = init_detector(config_file, device=device)
# inference the demo image
inference_detector(model, 'demo/demo.jpg')
然後運行：
python demo/demo.py
