MMDetection（一）：安装

简介

MMDetection 是一个基于 PyTorch 的目标检测开源工具箱。它是 OpenMMLab 项目的一部分。

主分支代码目前支持 PyTorch 1.5 以上的版本。

网站：https://github.com/open-mmlab/mmdetection

主要特性

• 模块化设计
  MMDetection 将检测框架解耦成不同的模块组件，通过组合不同的模块组件，用户可以便捷地构建自定义的检测模型。

• 丰富的即插即用的算法和模型
  MMDetection 支持了众多主流的和最新的检测算法，例如 Faster R-CNN，Mask R-CNN，RetinaNet 等。

• 速度快
  基本的框和 mask 操作都实现了 GPU 版本，训练速度比其他代码库更快或者相当，包括 Detectron2, maskrcnn-benchmark 和 SimpleDet。

• 性能高
  MMDetection 这个算法库源自于 COCO 2018 目标检测竞赛的冠军团队 MMDet 团队开发的代码，我们在之后持续进行了改进和提升。

安装

• 依赖：
  Linux 和 macOS （Windows 理论上支持）
  Python 3.6+
  PyTorch 1.3+
  CUDA 9.2+ （如果基于 PyTorch 源码安装，也能够支持 CUDA 9.0）
  GCC 5+
• 配置环境1
  安装pytorch和torchvision
  访问网站：https://pytorch.org/get-started/previous-versions/
  安装方法例如（cuda10.2安装pytorch1.9.0和torchvision0.10.0）：

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

• 配置环境2

 pip install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .

• 安装mmdetection

 pip install openmim
mim install mmdet

MIM 能够自动地安装 OpenMMLab 的项目以及对应的依赖包。

• 配置例子
  cuda 10.2
  python 3.7.11
  pytorch 1.10.2
  torchvision 0.11.3
  mmdetection 2.25.1
  mmcv-full 1.3.17