基于特征注意力的快速非均匀雾图像去雾算法
DOI:
CSTR:
作者:
作者单位:

1.三峡大学计算机与信息学院;2.湖北省水电工程智能视觉监测重点实验室 三峡大学计算机与信息学院 水电工程视觉监测宜昌市重点实验室宜昌

作者简介:

通讯作者:

中图分类号:

TP391.41

基金项目:

国家自然科学基金项目(面上项目,重点项目,重大项目)


Fast nonhomogeneous image dehazing algorithm based on feature attention
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有单幅图像去雾算法在雾度分布不均匀情况下去雾效果的局限性,以及较高的时间成本等问题。本文提出了一种以编码器-解码器结构为基本框架,融合多尺度卷积与特征注意力的快速图像去雾算法。首先,基于对性能和内存存储的权衡,使用了一种轻量级的编码器-解码器结构,以保证较低的时间成本;其次,考虑到在不均匀雾度分布的场景中,雾化的区域可能超过卷积核的大小,提出了一种多尺度卷积结构,在特征提取阶段首先使用1×1,3×3,5×5,7×7的并行多尺度卷积提取特征,以增大感受野保留输入图像的更多细节;此外,由于图像的不同空间域可以受到不同级别的雾霾的影响,不同场景深度的物体的退化程度也有很大差异,因此引入特征注意力基本块模块,有区别地对待不同通道和像素上的特征。为了验证算法的有效性,在三种数据集上将本文提出的算法与目前流行的算法进行对比实验。算法在真实的非均匀雾度分布场景NH-HAZE 2数据集中,三项评价指标峰值信噪比(PSNR)、结构相似性(SSIM)、单张分辨率为1600×1200的图片处理时间分别为20.50 dB、0.84、0.0038秒,相对于DMPHN模型均有所提高,三项指标皆处于较高水平。同时,在均匀雾度分布场景O-HAZE数据集中,也有着良好的表现。实验结果表明,本文算法有效地解决了在雾度分布不均匀情况下去雾效果不理想的问题,降低了时间成本,同时复原图像在颜色、亮度方面具有更好的表现。

    Abstract:

    Aiming at the limitations of the existing single image defogging algorithm in the case of uneven fog distribution and high time cost. In this paper, a fast image defogging algorithm based on encoder-decoder structure is proposed, which integrates multi-scale convolution and feature attention. Firstly, a lightweight encoder-decoder structure is used based on performance and memory storage trade-offs to ensure a low time cost. Secondly, considering that in the scene with uneven fog distribution, the atomized area may exceed the size of the convolution kernel, a multi-scale convolution structure is proposed. In the feature extraction stage, the parallel multi-scale convolution of 1×1, 3×3, 5×5, 7×7 is firstly used to extract features, so as to enlarge the sensitivity field and preserve more details of the input image. In addition, since different spatial domains of images can be affected by different levels of haze, the degradation degree of objects at different scene depths also varies greatly. Therefore, the basic block module of feature attention is introduced to treat the features on different channels and pixels differently. In order to verify the effectiveness of the algorithm, the proposed algorithm is compared with the current popular algorithms in three data sets. In the real NH-HAZE 2 data set with non-uniform haze distribution, the processing time of three evaluation indexes, namely, peak signal-to-noise ratio (PSNR), structural similarity (SSIM) and single image resolution of 1600×1200, are 20.50 dB, 0.84 and 0.0038 seconds, respectively. Compared with DMPHN model , all the three indexes are at a higher level. At the same time, it also has a good performance in O-HAZE data set with uniform haze distribution. Experimental results show that the proposed algorithm can effectively solve the problem of unsatisfactory fog effect in the case of uneven fog distribution, and reduce the time cost. Meanwhile, the restored image has a better performance in color and brightness.

    参考文献
    相似文献
    引证文献
引用本文
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-04-06
  • 最后修改日期:2023-06-19
  • 录用日期:2023-06-20
  • 在线发布日期:
  • 出版日期:
文章二维码