福州大学学报(自然科学版)2024,Vol.52Issue(1) :7-13.DOI:10.7631/issn.1000-2243.23099

注意力机制下的多层次特征融合图像去雾算法

Image dehazing algorithm based on multi-level feature fusion and attention mechanism

连文杰 叶少珍
福州大学学报(自然科学版)2024,Vol.52Issue(1) :7-13.DOI:10.7631/issn.1000-2243.23099

注意力机制下的多层次特征融合图像去雾算法

Image dehazing algorithm based on multi-level feature fusion and attention mechanism

连文杰 1叶少珍2
扫码查看

作者信息

  • 1. 福州大学计算机与大数据学院,福建 福州 350108
  • 2. 福州大学计算机与大数据学院,福建 福州 350108;福州大学智能制造仿真研究院,福建 福州 350108
  • 折叠

摘要

提出一种基于注意力机制的多层次特征融合的图像去雾算法.该算法通过残差密集网络和自校准卷积网络来提取不同尺度的特征,再利用双重注意单元和像素注意力将特征融合重建.同时采用一种由均方误差损失、边缘损失和鲁棒性损失函数相结合的损失函数,可以更好地保留细节特征.实验表明,该算法与其他去雾算法相比在峰值信噪比和结构相似度指标上得到一定的提高,去雾图像在主观视觉上取得了较好表现.

Abstract

An image dehazing network based on attention mechanism and multi-level feature fusion is proposed.The algorithm extracts features of different scales through residual dense network and self-calibration convolution network,and then uses dual attention units and pixel attention to fuse and reconstruct features.At the same time,a loss function combining mean square error loss,edge loss and robustness loss function is adopted,which can better retain the detail features.The experimental results show that this algorithm has made certain improvement in the peak signal to noise ratio and structural similarity indicators compared with other defogging algorithms,and the defogging image has made better performance in subjective vision.

关键词

图像去雾/注意力机制/多层次/特征融合

Key words

image dehazing/attention mechanism/multi-level/feature fusion

引用本文复制引用

基金项目

国家自然科学青年基金资助项目(41801324)

福建省自然科学基金资助项目(2019J01244)

福耀玻璃智能制造仿真平台资助项目(01001701)

出版年

2024
福州大学学报(自然科学版)
福州大学

福州大学学报(自然科学版)

CSTPCD北大核心
影响因子:0.35
ISSN:1000-2243
参考文献量16
段落导航相关论文