摘要

本文尝试改进了新的注意力，使用空间注意力和多轴频域注意力融合改进。改进后的注意力超越了GAM、BAM和CBAM等常用的注意力。

GAM

# 导入PyTorch的神经网络模块  
import torch.nn as nn  
# 导入PyTorch库  
import torch  
  
# 定义一个名为GAM_Attention的类，继承自nn.Module  
class GAM_Attention(nn.Module):  
    # 初始化函数，定义模型参数  
    def __init__(<