MMdetection 安装

安装环境： ubuntu20.04 、3070

1、首先需要创建一个虚拟环境，官方建议要 Python 3.7 以上，CUDA 9.2 以上和 PyTorch 1.8 及其以上

        创建一个python3.8的环境：conda create -n mmdetection python=3.8

       在创建的环境中安装pytorch：

# CUDA 11.6
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

2、安装MMDetection:

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements.txt

3、验证：

           mim download mmdet --config rtmdet_tiny_8xb32-300e_coco --dest .

出现问题：RuntimeError: mmdet is not installed. Please install it first.

解决：mim install mmdet
在当前文件夹中发现两个文件 rtmdet_tiny_8xb32-300e_coco.py 和 rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth。

          

运行以下命令进行验证：

python demo/image_demo.py demo/demo.jpg rtmdet_tiny_8xb32-300e_coco.py --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --device cpu

在当前文件夹中的 outputs/vis 文件夹中看到一个新的图像 demo.jpg，图像中包含有网络预测的检测框。

安装和验证完毕，开始学习。