MMSegmentation安装配置
一 基础环境配置
首先安装pytorch
pip3 install 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
然后 用 mim 安装 MMCV
以上步骤都没有爆红,接着以下的操作: 安装其他所需要的库
pip3 install opencv-python pillow matplotlib seaborn tqdm pytorch-lightning 'mmdet>=3.0.0rc1' -i https://pypi.tuna.tsinghua.edu.cn/simple
二 安装MMSegmentation
首先检查本地有没有MMSegmentation,有的话删除
# 删掉原有的 mmsegmentation 文件夹(如有)
rm -rf mmsegmentation
然后从官网克隆
下载完成之后我们进入刚下载的这个文件夹里面,编译它:(克隆了很多次,老被强)
下载好后,可以通过下面的指令检查是否安装成功:
三 使用MMSegmentation
我们可以在这里使用预训练好的模型和权重文件
Model Zoo:https://github.com/open-mmlab/mmsegmentation/blob/master/docs/en/model_zoo.md
这里可以下载他的配置文件和预训练模型,很方便的可以套用。
我们用PSPNet做实验