SENet(Squeeze-and-Excitation Networks)是2017年提出的一种经典的通道注意力机制,这种注意力可以让网络更加专注于一些重要的featuremap,它通过对特征通道间的相关性进行建模,把重要的特征图进行强化来提升模型的性能。论文链接 代码实现 模型 SENet的模型结果如图所示: 网络的输入 X X X是个多通道的图片,其维度为 [ H ′ ,
通道注意力机制SENet Diagram of a Squeeze-and-Excitation building block 实现代码如下 import torchimport torch.nn as nnclass SELayer(nn.Module):def __init__(self, channel, reduction=16):super(SELayer, sel