安装mmsegmentation默认主分支main

可见一班2024-04-22 2:01

安装时间2024.4.21

mmsegmentation新版本main分支（v1.2.2）

安装过程

python 复制代码

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

// 很关键，可以避免mmcv版本问题
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

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

git clone -b main https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .
# '-v' 表示详细模式，更多的输出
# '-e' 表示以可编辑模式安装工程，
# 因此对代码所做的任何修改都生效，无需重新安装

验证和测试

python 复制代码

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

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_4xb2-40k_cityscapes-512x1024.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --out-file result.jpg

将在当前文件夹中看到一个新图像 result.jpg，其中所有目标都覆盖了分割 mask

最大的问题在于mmcv，如果需要安装时，或者更改版本时，建议先卸载

pip uninstall mmcv
pip uninstall mmcv-full