【mmsegmentation】Windows10,11安装mmsegmentation

1 环境配置

需要先安装anaconda，CUDA11.3和cudnn

2 安装步骤

如果用git克隆代码失败，可以直接在github下载代码的压缩包也是一样的

conda create -n open-mmlab python=3.10 -y
conda activate open-mmlab

conda install pytorch=1.11.0 torchvision cudatoolkit=11.3 -c pytorch
set PATH=fullpath/mmsegmentation;%PATH%
pip install mmcv

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

2 测试代码

把路径切换到工程目录mmsegmentation，下载对应的pspnet的模型，下载地址，里面第一个模型

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --out-file result.jpg