Efficient Pyramid Split Attention金字塔分割注意力机制,YOLOv5改进之PSA注意力机制

目录

一、理论部分

网络结构

二、应用到YOLOv5

代码段

yaml文件

psac3



一、理论部分

最近,研究表明,通过在深度卷积神经网络中嵌入注意力模块,可以有效地提高其性能。本文提出了一种新的轻量级、高效的注意力方法——金字塔分割注意力模块。通过将ResNet瓶颈块中的3x3卷积替换为PSA模块,得到了一种新的表征块——高效金字塔分割注意(Efficient Pyramid Split Attention, EPSA)。EPSA模块可以作为即插即用组件轻松添加到已建立的骨干网络中,并且可以实现模型性能的显着改进。因此,在这项工作中,通过堆叠这些resnet风格的EPSA块,开发了一个简单高效的骨干架构EPSANet。相应地,本文提出的EPSANet可以为各种计算机视觉任务提供更强的多尺度表示能力,包括但不限于图像分类、目标检测、实例分割等。没有花哨的东西,所提出的EPSANet的性能优于大多数最先进的频道注意方法。与SENet-50相比,在ImageNet数据集上Top-1的准确率提高了1.93%,在MS-COCO数据集上使用mask - rcnn的目标检测精度提高了+2.7 box AP,在实例分割方面提高了+1.7 mask AP

网络结构

提出了一种新型的金字塔分散注意力模块。如图3所示,PSA模块主要分为四个步骤实现。首先,通过实现所提出的分割和拼接(SPC)模块,获得通道方向上的多尺度特征映

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

这糖有点苦

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值