MobileNet 改进:添加SpatialGroupEnhance 增加语义特征学习能力
目录
1. SpatialGroupEnhance 模块
2. MobileNetV2 + SpatialGroupEnhance 模块
3. 完整代码
Tips:融入模块后的网络经过测试,可以直接使用,设置好输入和输出的图片维度即可
1. SpatialGroupEnhance 模块
SpatialGroupEnhance
(SGE)是一种用于增强卷积神经网络(CNN)特征表示的模块,通常用于计算机视觉任务,如分类、检测和分割。SGE 通过显式地建模特征图中的空间关系,增强模型对局部特征的感知能力,从而提高模型的性能。
核心思想:
SGE 的核心思想是通过对特征图进行分组&#x