SE模块在RE-RESNET求和操作之前对面
向特征图进行了一个缩放操作,以减少特征图的尺寸,从而减少计算量。具体来说,SE模块包含两个步骤:
-
Global Average Pooling (GAP):对输入的特征图进行全局平均池化,将每个通道的特征值求平均得到一个标量值。
-
Scaling Operation:对每个通道的标量值进行一个缩放操作,使用两个全连接层来学习缩放因子,最终将缩放因子应用到对应通道的特征图上,从而得到缩放后的特征图。
这个缩放操作的效果是对不同的通道赋予不同的权重,使得重要的特征得到更多的关注,从而提高模型性能。在RE-RESNET求和操作之前,SE模块的输出与RE-RESNET的输入进行元素级别的乘法操作,从而得到最终的特征图。
原文地址: http://www.cveoy.top/t/topic/bzvl 著作权归作者所有。请勿转载和采集!