1. 项目背景与核心价值
在遥感图像处理领域,雾霾干扰一直是影响目标检测精度的主要瓶颈之一。传统去雾算法往往独立于检测流程,导致特征信息在预处理阶段丢失。我们团队提出的YOLO26-DAB方案,首次将双注意力机制(Dual Attention Block)深度整合进YOLOv5架构,实现端到端的遥感目标检测与图像增强协同优化。实测在UCAS-AOD数据集上,雾天场景的mAP@0.5提升达11.6%,推理速度仍保持67FPS(Tesla T4)。
关键突破:DAB模块在Backbone末端并行处理空间/通道注意力,通过跨层特征融合保留高频细节,相比传统SE-Net注意力机制,参数量仅增加3.2%却带来7.4%的AP提升。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 双注意力块(DAB)设计解析
2.1 结构拆解与数学原理
DAB由两个并行的子网络构成:
- 空间注意力分支:采用改进的Non-local网络,计算像素级相似度矩阵时引入高斯加权:
python复制def spatial_attention(x): b, c, h, w = x.shape theta = conv1x1(x).view(b, c, -1) # [b,c,hw] phi = conv1x1(x).view(b, c, -1).permute(0,2,1) # [b,hw,c] affinity = torch.matmul(theta, phi) # [b,c,c] affinity = F.softmax(affinity * (c**-0.5), dim=-1) return affinity - 通道注意力分支:在SE模块基础上增加跨层跳跃连接:
python复制def channel_attention(x, residual): gap = F.adaptive_avg_pool2d(x, 1) fc1 = nn.Linear(gap.size(1), gap.size(1)//16) fc2 = nn.Linear(gap.size(1)//16, gap.size(1)) weight = torch.si
