如何使用mmdetection训练实例分割模型？

入股安装 anoconda

从官方网站下载并安装。

配置环境

conda create --name openmmlab python=3.8 -y
conda activate mmdet

安装 PyTorch

网址：https://pytorch.org/get-started/previous-versions/

注意：这个步骤很关键，否则后面会出问题。一定要确保自己电脑当前安装和配置的cuda版本。

使用命令：nvcc -V 查看cuda版本

因为我电脑安装的是12.4，所以我选择pytorch时候，也要安装对应的。

如果你安装的是11.7，那么，就按照下图提示，选择以前的版本。

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

安装 MMEngine 和 MMCV

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

安装 MMDetection

从源码安装它：

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-v" 指详细说明，或更多的输出
# "-e" 表示在可编辑模式下安装项目，因此对代码所做的任何本地修改都会生效，从而无需重新安装。

选择配置文件并修改配置

【太简单，就暂时省略】

启动训练

启动命令：

python tools/train.py  configs/mask_rcnn/mask-rcnn_x101-32x4d_fpn_ms-poly-3x_coco.py

bug解决

如果报错：AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv>=2.0.0rc4, <2.1.0.

此处有一个终极解决方案【强烈推荐】：
安装 MMCV — mmcv 文档
比如。对于我的linux系统，我选择后就是下列的命令：

pip install mmcv==2.0.0rc3 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html

Traceback (most recent call last):
  File "tools/train.py", line 10, in <module>
    from mmdet.utils import setup_cache_size_limit_of_dynamo
  File "/home/uvtec-927/桌面/Projects/mmdetection/mmdet/__init__.py", line 16, in <module>
    assert (mmcv_version >= digit_version(mmcv_minimum_version)
AssertionError: MMCV==2.0.0rc3 is used but incompatible. Please install mmcv>=2.0.0rc4, <2.2.0.

如果还报错如上，那么：

pip install mmcv==2.0.0rc4 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html

我在windows上的解决方法是如下【对于linux不生效，还得用终极办法】：

pip install mmcv==2.0.0rc4