Windows下面安装MMDetection图文教程

1. https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md
2. 注意事项
   • 在安装MMDetection之前需要先将环境中的MMCV卸载掉
     • 注意：如果你安装了mmcv，你需要先运行 pip uninstall mmcv将其卸载掉。如果 mmcv 和 mmcv-full 都安装了，就会有ModuleNotFoundError.
   • MMCV的版本需要和PyTorch以及cuda的版本匹配才行
     • https://github.com/open-mmlab/mmcv#installation
   • 不用安装最新版的MMCV
     • 最新版的MMCV，MMDetection不支持
3. 安装要求
   • Linux or macOS (Windows is in experimental support)
   • Python 3.6+
   • PyTorch 1.3+
   • CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)
   • GCC 5+
   • MMCV
   • MMDetection和MMCV对应的版本要求如下
   • 
     
     • MMDetection_version MMCV_version master mmcv-full>=1.3.14, <1.4.0 2.18.0 mmcv-full>=1.3.14, <1.4.0 2.17.0 mmcv-full>=1.3.14, <1.4.0 2.16.0 mmcv-full>=1.3.8, <1.4.0 2.15.1 mmcv-full>=1.3.8, <1.4.0 2.15.0 mmcv-full>=1.3.8, <1.4.0 2.14.0 mmcv-full>=1.3.8, <1.4.0 2.13.0 mmcv-full>=1.3.3, <1.4.0 2.12.0 mmcv-full>=1.3.3, <1.4.0 2.11.0 mmcv-full>=1.2.4, <1.4.0 2.10.0 mmcv-full>=1.2.4, <1.4.0 2.9.0 mmcv-full>=1.2.4, <1.4.0 2.8.0 mmcv-full>=1.2.4, <1.4.0 2.7.0 mmcv-full>=1.1.5, <1.4.0 2.6.0 mmcv-full>=1.1.5, <1.4.0 2.5.0 mmcv-full>=1.1.5, <1.4.0 2.4.0 mmcv-full>=1.1.1, <1.4.0 2.3.0 mmcv-full==1.0.5 2.3.0rc0 mmcv-full>=1.0.2 2.2.1 mmcv==0.6.2 2.2.0 mmcv==0.6.2 2.1.0 mmcv>=0.5.9, =0.5.1,
4. 准备环境
   • conda create -n MMDetection python=3.7 -y
   • conda activate MMDetection
5. 安装PyTorch
   • MMDetection 是以PyTorch为底层框架的，所以需要首先安装PyTorch才行
   • 这个自己直接就是安装的最新的稳定版，我感觉应该会有很多特性支持
   • conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
   • 如果conda走不通的话，可以考虑使用pip安装
   • pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
6. 安装MMCV
   • Windows下源码安装
   • 参考这篇教程
   • https://zhuanlan.zhihu.com/p/427622928
     
7. 安装MMDetection
   • 一行代码的事
   • pip install mmdet
8. 安装额外的依赖项
   • 这个其实安不安装，对于一般问题而言区别不大，不过最好还是安装上去，省心
   • # for instaboost pip install instaboostfast # for panoptic segmentation pip install git+https://github.com/cocodataset/panopticapi.git # for LVIS dataset pip install git+https://github.com/lvis-dataset/lvis-api.git # for albumentations pip install albumentations>=0.3.2 --no-binary imgaug,albumentations
9. 验证安装
   • from mmdet.apis import init_detector, inference_detector config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' # download the checkpoint from model zoo and put it in `checkpoints/` # url: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth' device = 'cuda:0' # init a detector model = init_detector(config_file, checkpoint_file, device=device) # inference the demo image inference_detector(model, 'demo/demo.jpg')
   • 若是运行出下面的界面就可以了
10. 报错
    • ModuleNotFoundError: No module named 'mmcv._ext'
      • ModuleNotFoundError: No module named ‘mmcv._ext‘_一千零一夜的博客-CSDN博客
      • 使用whl安装
        • 不过Win平台的模块比较少，而且Python都是1.6版本的
        • https://download.openmmlab.com/mmcv/dist/index.html
      • 不过需要注意的是，找到对应的平台版本，否则会报下面的问题
        • 这个就是自己安装了Linux平台的代码报的错误
      • 简单来说就是MMCV没有安装好，按照上面的教程步骤再走一遍吧。
12.