高级检索
陈志华, 杜垣均, 温阳, 盛斌, 王继红, 毛丽娟, 吴恩华. 基于注意力机制的密集残差融合与空间局部滤波低光照去雾算法[J]. 计算机辅助设计与图形学学报, 2022, 34(12): 1842-1849. DOI: 10.3724/SP.J.1089.2022.19188
引用本文: 陈志华, 杜垣均, 温阳, 盛斌, 王继红, 毛丽娟, 吴恩华. 基于注意力机制的密集残差融合与空间局部滤波低光照去雾算法[J]. 计算机辅助设计与图形学学报, 2022, 34(12): 1842-1849. DOI: 10.3724/SP.J.1089.2022.19188
CHEN Zhi-hua, DU Yuan-jun, WEN Yang, SHENG Bin, WANG Ji-hong, MAO Li-juan, WU En-hua. Dense Residual Fusion and Spatial Local Filtering Low Light Dehazing Algorithm Based on Attention Mechanism[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(12): 1842-1849. DOI: 10.3724/SP.J.1089.2022.19188
Citation: CHEN Zhi-hua, DU Yuan-jun, WEN Yang, SHENG Bin, WANG Ji-hong, MAO Li-juan, WU En-hua. Dense Residual Fusion and Spatial Local Filtering Low Light Dehazing Algorithm Based on Attention Mechanism[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(12): 1842-1849. DOI: 10.3724/SP.J.1089.2022.19188

基于注意力机制的密集残差融合与空间局部滤波低光照去雾算法

Dense Residual Fusion and Spatial Local Filtering Low Light Dehazing Algorithm Based on Attention Mechanism

  • 摘要: 低光照场景的雾霾图像在去雾过程中易产生颜色失真、斑块和伪影等现象,针对此问题,提出一种适用于低光照场景的基于注意力机制的密集残差融合与空间局部滤波去雾算法.首先利用密集残差块增加神经网络深度,使网络学习更高级的特征信息;然后引入空间与通道注意力机制对特征进行过滤和筛选,使网络可以区分光照不均匀区域,解决颜色失真等问题;采用空间局部滤波增强的方法,提高去雾结果的对比度、清晰度和能见度;最后设计了联合损失函数约束网络的学习,避免串联结构的误差放大以及学习混合退化.在PyTorch环境下,用夜间城市合成雾霾数据集NHR进行测试,并与现有的FFANet,GridDehaze等去雾算法进行对比.实验结果表明,与其他去雾算法相比,所提算法的峰值信噪比提升8.01~14.16dB,结构相似度提高0.10~0.36.所提算法还解决了颜色失真、斑块和伪影等问题.

     

    Abstract: Image dehazing under low-light condition is easy to produce color distortion,patches and artifacts.To adapt to low-light condition,a dense residual fusion and spatial local filtering dehazing algorithm based on attention mechanism is proposed.Firstly,dense residual block is proposed to increase the depth of the neural network, so that the network can learn more advanced feature information. Then, the spatial and channel attention mechanism is introduced to filter and screen the features, so that the network can distinguish the uneven illumination areas and solve the problems of color distortion. The model adopts the method of spatial local filter enhancement to improve the contrast, clarity and visibility of dehazing results. Finally, a learning of joint loss function constraint network is designed to avoid error amplification and learning hybrid degradation of series structure. In the PyTorch environment, the night urban synthetic haze dataset (NHR) is used for testing and compared with the existing dehazing algorithms such as FFANet and GridDehaze. The experimental results show that compared with other dehazing algorithms, proposed method improves PSNR by 8.01-14.16 dB and increases SSIM by 0.10-0.36. In addition, proposed method solves the problems of color distortion, plaque and artifact.

     

/

返回文章
返回