1. GridDehazeNet论文精读:端到端可训练的图像去雾网络架构解析
在计算机视觉领域,图像去雾(Image Dehazing)一直是低层视觉任务中的研究热点。传统方法通常基于大气散射模型进行物理建模,而近年来深度学习技术的引入使得端到端的去雾网络成为可能。GridDehazeNet作为2019年发表在ACM Multimedia会议上的创新工作,通过独特的网格状连接结构和注意力机制,在NTIRE 2019图像去雾挑战赛中取得了优异成绩。本文将深入解析该网络的架构设计、核心组件以及实际应用效果。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 网络架构与核心创新点
2.1 整体网络结构设计
GridDehazeNet采用编码器-解码器结构,其创新之处在于引入了密集网格状连接(Dense Grid Connection)机制。与传统的U-Net结构不同,该网络在编码器和解码器之间建立了跨层、跨尺度的密集连接,形成了类似网格的拓扑结构。这种设计带来了三个显著优势:
- 多尺度特征融合:不同层级的特征能够直接交互
- 梯度传播优化:缓解了深层网络的梯度消失问题
- 信息流动增强:低频和高频信息得到充分混合
网络输入为512×512的雾图,经过5个下采样阶段后,通过对称的上采样路径恢复分辨率。每个阶段包含2-3个残差块,中间通过最大池化进行降采样。
2.2 关键组件技术解析
2.2.1 通道注意力模块(CAM)
在解码器每个阶段引入通道注意力机制,其实现流程为:
- 全局平均池化生成通道统计量
- 两层全连接层学习通道间关系
- Sigmoid激活生成注意力权重
- 原始特征与权重逐通道相乘
该模块使网络能够自适应地强调重要特征通道,抑制噪声干扰,在PSNR指标上带来了约0.3dB的提升。
2.2.2 密集网格连接
连接方式遵循以下规则:
- 第i个编码器块连接到第(5-i)个解码器块
- 相邻尺度间建立跳跃连接
- 所有连接采用1×1卷积统一通道数
这种连接模式使得浅层的细节特征能够直接指导深层语义特征的恢复,在主观质量评价中显著改善了边缘锐度。
3. 训练策略与实现细节
3.1 损失函数设计
网络采用多尺度复合损失函数:
- L1重建损失:保证像素级精度
$$ \mathcal{L}_{rec} =
