MMsegmentation安装教程

MM Segmentation

MMSegmentation 是 OpenMMLab 开源的基于 PyTorch 实现的功能强大的语义分割工具箱

MMSegmentation 可以在 Linux、Windows 和 MacOS 上运行。它需要 Python 3.6 以上，CUDA 9.2 以上和 PyTorch 1.3 以上

MM Segmentation安装

https://mmsegmentation.readthedocs.io/en/latest/get_started.html

1、使用MIM安装MMCV

pip install -U openmim
mim install mmcv-full

2、安装 MMSegmentation

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

3、验证安装

下载config和checkpoint文件

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

验证demo文件

python3 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

测试成功便会输出result.jpg