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

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

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

image dehazingattention mechanismmulti-levelfeature fusion

连文杰、叶少珍

展开 >

福州大学计算机与大数据学院,福建 福州 350108

福州大学智能制造仿真研究院,福建 福州 350108

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

国家自然科学青年基金资助项目福建省自然科学基金资助项目福耀玻璃智能制造仿真平台资助项目

418013242019J0124401001701

2024

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

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

CSTPCD北大核心
影响因子:0.35
ISSN:1000-2243
年,卷(期):2024.52(1)
  • 16