一种基于FFA-Net改进的单幅图像去雾算法

作     者：何钦 徐望明 王义焕 罗扬 王薇 HE Qin;XU Wangming;WANG Yihuan;LUO Yang;WANG Wei

作者机构：武汉科技大学信息科学与工程学院湖北武汉430081 武汉科技大学冶金自动化与检测技术教育部工程研究中心湖北武汉430081 武汉科技大学计算机科学与技术学院湖北武汉430065 

基　　金：国家自然科学基金项目(62202347) 湖北省中央引导地方科技发展专项(2023EGA001) 武汉科技大学冶金自动化与检测技术教育部工程研究中心开放课题(MADTOF2021B02) 

出 版 物：《武汉科技大学学报》 (Journal of Wuhan University of Science and Technology)

年 卷 期：2024年第47卷第6期

页      码：448-456页

摘      要：雾霾会严重影响使用卷积神经网络的视觉系统对目标图像的检测和识别能力,为此本文在特征融合注意力网络FFA-Net的基础上设计和添加全局空间上下文增强(GSCE)模块和细节渐进增强(PDE)模块,进而提出一种改进型单幅图像去雾算法。GSCE模块用于增强全局空间信息,PDE模块用于逐步细化和增强图像特征,二者结合进行高效和轻量级的特征提取,弥补原模型中大量使用跳跃连接所造成的细节信息损失。改进模型分别在公共基准数据集RESIDE的室内数据和室外数据上进行训练,并分别在SOTS的室内和室外两个数据集上进行了测试。结果表明,本文算法明显超越了原FFA-Net和现有典型的单幅图像去雾算法,尤其在SOTS室内测试数据集上,单独融合GSCE模块就使得PSNR指标从36.36 dB提升到38.39 dB,在进一步使用PDE模块后PSNR指标提升到38.78 dB,算法的去雾性能得到较大提高,验证了改进策略的有效性。

主 题 词：图像去雾 卷积神经网络 FFA-Net 全局空间上下文增强模块 细节渐进增强模块 

学科分类：1305[艺术学-设计学类] 13[艺术学] 081104[081104] 08[工学] 0804[工学-材料学] 081101[081101] 0811[工学-水利类] 

D　O　I：10.3969/j.issn.1674-3644.2024.06.006

馆 藏 号：203155892...