摘要 本文尝试改进了新的注意力,使用空间注意力和多轴频域注意力融合改进。改进后的注意力超越了GAM、BAM和CBAM等常用的注意力。 GAM # 导入PyTorch的神经网络模块 import torch.nn as nn # 导入PyTorch库 import torch # 定义一个名为GAM_Attention的类,继承自nn.Module class GAM_Attention(nn.Module)