YoloV8改进策略:改进Head|自研频域和空间注意力,超越GAM,CBAM等注意力|注意力创新改进|高效涨点|代码注释与改进|包括改进后的结构图

摘要

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

GAM

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

猜你喜欢

转载自blog.csdn.net/m0_47867638/article/details/135785913