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

安装 anoconda

从官方网站下载并安装。

配置环境

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

安装 PyTorch

注意：这个步骤很关键，否则后面会出问题。一定要确保自己电脑当前安装和配置的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.

解决方法：

pip install mmcv==2.0.0rc4