mmsegmentation进阶使用学习笔记（二）——添加自定义模块

一、开发新组件

1、添加新的主干网络（backbone）

所有组件，包括主干网络（backbone）、头、损失函数、数据预处理器、分割器（segmentor）等都可以自定义。

（1）添加主干网络

主干网络在深度学习模型中主要负责特征提取、生成多尺度特征表示。在语义分割中就对了编码器。

Step1：创建新的文件mmseg/models/backbones/gleformer.py

根据自己所需要的模型搭建backbone，我设计的并行编码器需要参考mit.py和unet.py

Step2：在mmseg/models/backbones/__init__.py 中引入模块

Step3：在配置文件中使用它