MMSegmentation 安装教程，避免mmcv不合适版本，及相关问题解决

某睿鸭

已于 2025-05-08 11:53:45 修改

阅读量328

点赞数 2

文章标签： python 语义分割安装 MMSegmentation

于 2025-03-05 17:50:26 首次发布

本文链接：https://blog.youkuaiyun.com/qq_43683133/article/details/146049022

版权

创建虚拟环境

conda create --name openmmlab python=3.10 -y
conda activate openmmlab

python 3.8/3.10 都行，不做要求

安装 torch

很重要，mmcv冲突的主要原因
作者这边的cuda版本是12.2

# CUDA 12.1
conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia

在这里插入图片描述

其他的安装

pip install -U openmim
mim install mmengine

MMCV 安装

pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html

在这里插入图片描述

从源码安装

git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .

其他需要安装的包

pip install "mmdet>=3.0.0rc4"    
pip install mmpretrain>=1.0.0rc7
pip install ftfy
pip install regex
pip install numpy==1.26.0

Test

第一步我们需要下载配置文件和模型权重文件。


mim download mmsegmentation --config pspnet_r50-d8_512x1024_40k_cityscapes --dest .

下载将需要几秒钟或更长时间，这取决于你的网络环境。完成后，你会在当前文件夹中发现两个文件pspnet_r50-d8_512x1024_40k_cityscapes.py和pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth。

第二步验证推理示例

如果您是从源码安装的 MMSegmentation，那么直接运行以下命令进行验证：

python demo/image_demo.py demo/demo.png pspnet_r50-d8_512x1024_40k_cityscapes.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cpu --out-file result.jpg

你会在你的当前文件夹中看到一个新的图像result.jpg，其中的分割掩膜覆盖在所有对象上。

如果以上步骤，没有办法解决你的冲突问题，建议你根据以下两个链接，搭配找到最适合你的版本

https://pytorch.org/get-started/previous-versions/#osx-32
https://mmcv.readthedocs.io/en/latest/get_started/installation.html

问题合集

在使用 cityscapes.py 文件时报错 `module 'mmcv' has no attribute 'mkdir_or_exist'`

将该文件里的所有 mmcv 替换成 mmengine

使用 `mim download mmsegmentation --dataset cityscapes` 报错

检查 mmseg/.mim/dataset-index.yml 路径是否存在
Command '['odl', 'get', 'OpenDataLab/CityScapes', '-d', 'data']' returned non-zero exit status 255. 修改为 subprocess.check_call(['openxlab','dataset', 'get','--dataset-repo', src_name, '--target-path', download_root], 修改的文件位置为 vim ~/miniconda3/envs/DNNTesting/lib/python3.1/site-packages/mim/commands/download.py

libGL.so.1: cannot open shared object file: No such file or directory

pip install opencv-python
pip install opencv-python-headless