安装时间2024.4.21
mmsegmentation新版本main分支(v1.2.2)
安装过程
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' 表示以可编辑模式安装工程,
# 因此对代码所做的任何修改都生效,无需重新安装
验证和测试
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