基于频率与注意力机制的图像去雾算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(41975183)


Image Dehazing Algorithm Based on Frequency and Attention Mechanism
Author:
Affiliation:

Fund Project:

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

    由于大气雾和气溶胶的存在, 图像能见度显著下降且色彩失真, 给高级图像识别带来极大困难. 现有的图像去雾算法常存在过度增强、细节丢失和去雾不充分等问题. 针对过度增强和去雾不充分的问题, 本文提出了一种基于频率和注意力机制的图像去雾算法(frequency and attention mechanism of the image dehazing network, FANet). 该算法采用编码器-解码器结构, 通过构建双分支频率提取模块获取全局和局部的高低频信息. 构建频率融合模块调整高低频信息的权重占比, 并在下采样过程中引入附加通道-像素模块和通道-像素注意力模块, 以优化去雾效果. 实验结果显示, FANet在SOTS-indoor数据集上的PSNR和SSIM分别为40.07 dB和0.9958, 在SOTS-outdoor数据集上分别为39.77 dB和0.9958. 同时, 该算法也在HSTS和Haze4k测试集上取得了不错的结果, 与其他去雾算法相比有效缓解了颜色失真和去雾不彻底等问题.

    Abstract:

    Atmospheric fog and aerosols can significantly reduce visibility and distort colors in images, bringing great difficulties to advanced image recognition. Existing image dehazing algorithms often face problems such as excessive enhancement, loss of details, and insufficient dehazing. To avoid excessive enhancement and insufficient dehazing, this study proposes an image dehazing algorithm based on frequency and attention mechanisms. The algorithm adopts an encoder-decoder structure and constructs a dual-branch frequency extraction module to obtain both global and local high and low-frequency information. A frequency fusion module is then constructed to adjust the weight proportions of the high and low-frequency information. To optimize the dehazing effect, the algorithm introduces an additional channel-pixel module and a channel-pixel attention module during down sampling. Experimental results show that FANet achieves a PSNR of 40.07 dB and an SSIM of 0.9958 on the SOTS-indoor dataset, and a PSNR of 39.77 dB and an SSIM of 0.9958 on the SOTS-outdoor dataset. The proposed algorithm also achieves good results on the HSTS and Haze4k test sets. It effectively alleviates color distortion and incomplete dehazing compared with other dehazing algorithms.

    参考文献
    相似文献
    引证文献
引用本文

王军,孟儒君,程勇.基于频率与注意力机制的图像去雾算法.计算机系统应用,,():1-10

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-06-19
  • 最后修改日期:2024-07-10
  • 录用日期:
  • 在线发布日期: 2024-11-28
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号