低版本MMSegmentation一次安装运行成功

MMSegmentation安装教程

机器环境 cuda11.7（cuda可以向下兼容，高版本cuda不影响安装）

1.创建环境

conda create -n mmseg python=3.8 -y

2.进入环境

conda activate mmseg

3.安装低版本的 pytorch==1.13, cuda11.7

pip install torch==1.13.0 torchvision --index-url https://download.pytorch.org/whl/cu117

4.安装mmcv==2.0.0rc4 安装 MMCV

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

5.下载mmsegmentation文件

文件下载

点击进入，下载zip包，解压就可以

6.配置环境

解压之后，在终端环境中进入文件夹，并启动相应的环境

安装必要的库

pip install -v -e . # 句尾有个点别忽略了

7. 安装mmengin

pip install mmengin==0.7.0

注意：在安装的过程中，如果出现 “No modules named xxx”，缺少相应的库

直接采用 : pip install xxx 

安装缺少的库就可以了

8. 验证测试

安装 mim 工具，下载相应的测试文件

pip install -U openmim

安装完成后下载 测试文件

mim download mmsegmentation --config pspnet_r50-d8_4xb2-40k_cityscapes-512x1024 --dest .

下载完成后是这样的

在安装过程中，会出现一个错误，需要把 yapf 库的版本降低到 0.40.1版本

具体为： pip uninstall yapf           # 先卸载

               pip install yapf==0.40.1 # 再安装

打开项目中的demo文件夹，找到 image_demo.py文件，配置以下四个主要参数

# 测试图片
parser.add_argument('--img', default='demo/demo.png', help='Image file')
# 测试模型配置文件  
parser.add_argument('--config', default='pspnet_r50-d8_4xb2-40k_cityscapes-512x1024.py', help='Config file')  
# 测试模型权重文件
parser.add_argument('--checkpoint', default='pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth', help='Checkpoint file')  
# 保存文件夹，保存文件为demo.png
parser.add_argument('--out-file', default='wordir/demo.png', help='Path to output file')

运行 image_demo.py

python demo/image_demo.py

结果在wordir文件夹，结果如下

安装测试成功！

可以打印mmsegmentation的版本

import mmseg
print(mmseg.__version__)

以上就是低版本MMSegmentation安装教程~