AI实战训练营&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做实验
请添加图片描述

猜你喜欢

转载自blog.csdn.net/shengweiit/article/details/131194053