CBAM
CBAM 模块概述 通道注意力模块(Channel Attention Mechanism)和空间注意力模块(Spatial Attention Mechanism)是注意力机制的两种主要形式,它们分别通过对通道维度和空间维度的特征图进行加权,从而使网络更加关注重要的特征。CBAM模块结合了这两种注意力机制,可以在保留空间信息的同时,有效地提取关键通道特征,提高了网络在处理复杂图像任务上的性能表现。
代码实现
import torch
from torch import nn
from torchsummary import summary
class ChannelModule(n