简介:SENET通道注意力机制源代码+注释 import torchimport torch.nn as nnclass SENET(nn.Module):def __init__(self, channel, r=0.5): # channel为输入的维度, r为全连接层缩放比例->控制中间层个数super(SENET, self).__init__()# 全局均值池化self.global_avg_pool = nn.AdaptiveAvgPool2d(1)# ...
代码地址:GitHub - hujie-frank/SENet: Squeeze-and-Excitation Networks Abstract—摘要 翻译 卷积神经网络是建立在卷积运算操作上,通过融合空间和通道级信息在局部感受野内提取信息特征。为促进网络表征力,最近的一些方法已展示一些增强空间编码的益处。在该工作中,我们聚焦在通道关系上并提出一种新的架构单元,我们称之...