一、本文介绍
本文记录的是利用PPA (并行补丁感知注意模块)改进RT-DETR检测精度,详细说明了优化原因,注意事项等。原论文在红外小目标检测任务中,小目标在多次下采样操作中容易丢失关键信息。PPA模块通过替代编码器和解码器基本组件中的传统卷积操作,更好地保留小目标的重要信息。
专栏目录:RT-DETR改进目录一览 | 涉及卷积层、轻量化、注意力、损失函数、Backbone、SPPF、Neck、检测头等全方位改进
文章目录
二、PPA 介绍
HCF-Net: Hierarchical Context Fusion Network for Infrared Small Object Detection
2.1 原理
2.1.1 多分支特征提取原理
采用多分支特征提取策略,通过不同分支提取不同尺度和层次的特征。利用局部、全局和串行卷积分支,对输入特征张量进行处理。通过控制 patch size参数实现局部和全局分支的区分,计算非重叠 patch之间的注意力矩阵,实现局部和全局特征提取与交互。在特征提取过程中,还通过一系列操作对特征进行选择和调整权重,最终将三个分支的结果求和得到融合后的特征。
2.1.2 特征融合和注意力原理
在多分支特征提取后,利用注意力机制进行自适应特征增强。注意力模块包括高效的通道注意力和空间注意力组件。首先通过一维通道注意力图和二维空间注意力图对特征进行依次处理,然后经过一系列激活函数、批归一化和 dropout等操作,得到最终输出。
2.2 结构
2.2.1 多分支特征提取结构
- 主要由多分支融合和注意力机制两部分组成。多分支融合部分包括 patch - aware和串联卷积。patch - aware中的参数
p设置为2和4,分别代表局部和全局分支。对于输入特征张量
订阅专栏 解锁全文
6万+





被折叠的 条评论
为什么被折叠?



