CBAM 注意力机制模块(通道+空间)
#(1)通道注意力机制 (相比较SE 多了全局最大池化)
class channel_attention(nn.Module):
# 初始化, in_channel代表输入特征图的通道数, ratio代表第一个全连接的通道下降倍数
def __init__(self, in_channel, ratio=4):
# 继承父类初始化方法
super(channel_attention, self).__init__()
# 全
原创
2024-01-03 15:25:49 ·
635 阅读 ·
1 评论