由于目前网上的mmsegmentation的安装教程比较老,有些步骤现在照着来会出现莫名其妙的报错,所以在这里整理一下自己这次安装的流程,供大家参考。
1. 安装anaconda
这里就不给教程了,网上很多
2. 建立环境
conda create --name mms python=3.8
3. 安装pytorch
这个网上教程也很多,记得安装对应版本
4. 使用MIM安装MMCV
pip install -U openmim
mim install mmengine
mim install mmcv==2.0.0
5. 安装 MMSegmentation
新建个文件夹,然后右键、git bash here,输入以下代码
git clone -b main https://github.com/open-mmlab/mmsegmentation.git
然后在anaconda Prompt里cd到刚才下载好的项目文件夹下面
输入以下代码(# 后为注释内容,不用复制上去),将该项目作为动态(也就是可修改的)的第三方库安装到你建立的环境里。
pip install -v -e .
# '-v' 表示详细模式,更多的输出
# '-e' 表示以可编辑模式安装工程,
# 因此对代码所做的任何修改都生效,无需重新安装
6. 测试部分
(1)下载配置文件和模型文件
mim download mmsegmentation --config pspnet_r50-d8_4xb2-40k_cityscapes-512x1024 --dest .
(2) 验证推理 demo
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
(3) 测试结果
在项目文件夹下会多出一个result.jpg,如下图所示:
大功告成。