YOLOv10改进 | 独家创新- 注意力篇 | YOLOv10引入结合ECA和NRMS形成全新的ERMS注意力机制和C2f_ERMS(全网独家创新)

1. ERMS介绍

          (1). 整合通道和空间注意力机制:
          ECA模块专注于通道间的交互,通过全局平均池化提取每个通道的全局信息,然后通过一维卷积生成每个通道的权重,最终使用Sigmoid激活函数得到通道注意力权重,从而增强重要通道的特征。然而,ECA模块没有考虑到空间信息的交互。
          NRMS模块专注于空间维度上的注意力机制,通过查询、键和值的线性变换,以及自注意力机制来捕捉空间位置间的相关性。同时,NRMS模块还可以通过相对位置编码进一步提升空间特征的提取能力。但NRMS模块没有考虑通道间的关系。
          ERMS模块则将ECA和NRMS模块的优势结合起来,既考虑了通道间的交互,也考虑了空间维度上的相关性,从而能够更全面地捕捉图像的特征信息。

          (2). 灵活的权重调整机制:
          ERMS模块通过引入两个可学习的参数α和β,分别控制ECA模块输出和NRMS模块输出在最终结果中的权重。这种灵活的权重调整机制使得ERMS模块能够在不同的任务和数据集上动态调整通道和空间注意力的贡献,进一步提升模型的表现。

          (3). 丰富的特征表达能力:
         ECA模块通过全局平均池化和一维卷积捕捉每个通道的全局信息,增强了重要通道的特征。
         NRMS模块通过多头自注意力机制,捕捉不同空间位置间的复杂关系,并且可以通过相对位置编码进一步提升空间特征的表达能力。
         ERMS模块结合了ECA和NRMS的优势,既能够增强重要通道的特征,又能捕捉空间位置间的复杂关系,从而提供更丰富和多样的特征表达能力。

          (4). 更强的特征选择能力:

  • 15
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值