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