安装detectron2
安装依赖库
-
使用 Python ≥ 3.6 的 Linux 或 macOS
-
PyTorch≥1.6和torchvision的PyTorch安装相匹配。将它们一起安装在pytorch.org以确保这一点
-
OpenCV 是可选的，但演示和可视化需要
pip install cython pycocotools
从源代码构建 Detectron2
需要 gcc & g++ ≥ 5.4。ninja是可选的，但推荐用于更快的构建。拥有它们后，运行：
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' # (add --user if you don't have permission) # Or, to install it from a local clone: git clone https://github.com/facebookresearch/detectron2.git python -m pip install -e detectron2
要重建从本地克隆构建的detectron2，请先使用清理旧构建。重新安装 PyTorch 后，您经常需要重建detectron2。rm -rf build/ **/*.so
安装预构建的 Detectron2（仅限 Linux）
从此表中选择安装v0.4（2021 年 3 月）：
| CUDA | pytorch 1.8 | pytorch 1.7 | pytorch 1.6 |
|---|---|---|---|
| 11.1 | 安装 | ||
| 11.0 | 安装 | ||
| 10.2 | 安装 | 安装
|
安装 |
| 10.1 | 安装 | 安装 | 安装 |
| 9.2 | 安装 | 安装 | |
| CPU | 安装 | 安装 | 安装 |
注意：
-
预建包必须与相应版本的 CUDA 和 PyTorch 的官方包配合使用。否则，请从源代码构建detectron2。
-
新软件包每隔几个月发布一次。因此，程序包在master分支中可能不包含最新功能，并且可能与使用detectron2的研究项目的master分支（例如，项目中的那些）不兼容。
安装mmdetection
安装依赖库
pip install numpy cython
安装mmcv-full
pip直接安装
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
源码编译安装
git clone https://github.com/open-mmlab/mmcv.git cd mmcv MMCV_WITH_OPS=1 pip install -e . # package mmcv-full will be installed after this step cd ..
安装mmdetection
git clone https://github.com/open-mmlab/mmdetection.git # or download this repo to local cd mmdetection python setup.py develop
everything is OK!!