MMSegmentation 代码实践——OpenMMLab第七讲

MMSegmentation
语义分割是对每一个像素进行分类

MMSegmentation 手册(同济子豪兄) https://github.com/TommyZihao/MMSegmentation_Tutorials/

A 安装配置

安装PyTorch
云GPU上安装，配置RTX3060、CUDA 11.2，会提示卸载torch1.10.0+cu113，成功安装torch1.10.1+cu113。

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

!pip install -U openmim
!mim install mmengine
# !mim install 'mmcv>=2.0.0rc1'
!mim install 'mmcv==2.0.0rc4'

此处选择mmcv==2.0.0rc4，而不是子豪note里的2.0.0rc3。

安装其他工具包

pip install opencv-python pillow matplotlib seaborn tqdm 'mmdet>=3.0.0rc1' -i https://pypi.tuna.tsinghua.edu.cn/simple

下载MMSegmentation

# 删掉原有的 mmtracking 文件夹（如有）
rm -rf mmsegmentation
# 从 github 上下载最新的 mmsegmentation 源代码
git clone https://github.com/open-mmlab/mmsegmentation.git -b dev-1.x

安装MMSegmentation
进入mmsegmentation目录，pip 安装。

cd mmsegmentation
pip install -v -e .

下载预训练模型权重文件和视频素材
创建checkpoint、outputs、data三个文件夹

mkdir checkpoint
mkdir outputs
mkdir data
# 下载预训练权重文件到checkpoint文件夹
wget https://download.openmmlab.com/mmsegmentation/v0.5/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth -P checkpoint
# 下载街景图片素材
wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220713-mmdetection/images/street_uk.jpeg -P data
# 上海驾车街景视频，视频来源：https://www.youtube.com/watch?v=ll8TgCZ0plk
!wget https://zihao-download.obs.cn-east-3.myhuaweicloud.com/detectron2/traffic.mp4 -P data
# 街拍视频，2022年3月30日
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220713-mmdetection/images/street_20220330_174028.mp4 -P data

检查安装成功

• 检查pytorch，输出“Pytorch版本1.10.1+cu113，CUDA是否可用 True”。

# 检查 Pytorch
import torch, torchvision
print('Pytorch 版本', torch.__version__)
print('CUDA 是否可用',torch.cuda.is_available())

• 检查mmcv， 输出“MMCV版本 2.0.0rc4，CUDA版本11.3, 编译器版本GCC 9.3”

import mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print('MMCV版本', mmcv.__version__)
print('CUDA版本', get_compiling_cuda_version())
print('编译器版本', get_compiler_version())

• 检查mmsegmentation， 输出“mmsegmentation版本 1.0.0rc5”

import mmseg
from mmseg.utils import register_all_modules
from mmseg.apis import inference_model, init_model
print('mmsegmentation版本', mmseg.__version__)

B 预训练语义分割模型预测——单张图像和视频

C 在自己的数据集上训练语义分割模型

准备训练文件
修改数据集类
修改config配置文件

开始训练

使用训练得到的模型预测图像