激光与光电子学进展, 2024, 61 (4): 0401002, 网络出版: 2024-02-22  

并行多尺度注意力映射图像去雾算法

Parallel Multi Scale Attention Mapping Image Dehazing Algorithm
作者单位
辽宁工程技术大学软件学院,辽宁 葫芦岛 125105
摘要
针对当前去雾算法中容易产生的图像伪影、图像颜色失真、图像细节模糊不清等问题,提出一种并行多尺度注意力映射图像去雾算法,通过端到端方式以编码器解码器结构实现图像去雾。在编码阶段,采用连续下采样层降低特征维度,避免过拟合。在特征转换阶段,采用并行分支结构设计并行多尺度注意力映射模块,使模型能够在关注图像重要特征的同时充分利用多尺度特征,并通过并行连接选择性特征融合模块有效收集图像空间结构信息。解码阶段,采用上采样层重构图像,并通过上下采样层融合更好地保留图像边缘信息。实验结果表明,该算法在合成雾天数据集以及真实雾天图像上均具有较好的去雾效果,相较于传统去雾算法,可更好地保留图像细节,具备较好的色彩保持度。
Abstract
Problems such as image color distortion, blurred image details, and image artifacts are prone to occur in the current dehazing algorithm. In order to solve the above problems, an image dehazing algorithm with parallel multi scale attention mapping is proposed. The algorithm achieves image defogging through an end-to-end encoder decoder structure. In the encoder stage, the continuous downsampling layer is used to reduce feature dimension and avoid over-fitting. In the feature transformation stage, a parallel multi scale attention mapping block with a parallel branch structure is designed, so that the model can make full use of multi scale features while focusing on important features of the image, and effective collection of image spatial structure information by connecting selective feature fusion block in parallel. In the decoding stage, the upsampling layer is used to reconstruct the image, and through skip connections of up and down sampling to better preserve image edge information. Experimental results show that the algorithm has better dehazing effects on both synthetic hazy datasets and real hazy images. Compared with traditional dehazing methods, this algorithm better preserves image details and has better color retention.

袁姮, 颜廷昊. 并行多尺度注意力映射图像去雾算法[J]. 激光与光电子学进展, 2024, 61(4): 0401002. Heng Yuan, Tinghao Yan. Parallel Multi Scale Attention Mapping Image Dehazing Algorithm[J]. Laser & Optoelectronics Progress, 2024, 61(4): 0401002.

引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!