多尺度特征融合和上下文混合注意力机制的单图像去模糊
作者机构:南京邮电大学自动化学院人工智能学院 闽西职业技术学院城乡建筑学院
出 版 物:《小型微型计算机系统》 (Journal of Chinese Computer Systems)
年 卷 期:2024年
学科分类:08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程]
主 题:图像去模糊 多尺度特征表示 上下文混合注意力 选择性特征融合 双层特征融合增强
摘 要:随着深度学习技术的快速发展,陆续提出了多种端到端的单图像去模糊网络。然而,现有的端到端网络在提高去模糊性能的同时,也带来了较高的计算复杂度。例如,传统的UNet网络虽然在图像去模糊任务中表现出色,但仍存在一些缺陷。首先,UNet网络的计算复杂度较高。其次,由于UNet网络的特征融合策略相对简单,导致其无法提取显著性的特征表征。为解决UNet网络的这些不足,本文提出多尺度特征融合和上下文混合注意力机制的单图像去模糊方法。首先,利用提出的多尺度特征表示机制快速提取不同分辨率的特征表示,从而降低计算复杂度。其次,利用提出的上下文混合注意力模块提取重要的空间信息和信道信息,利用提出的选择性特征融合模块提取自适应加权优化的空间信息和信道信息,利用提出的双层特征融合增强模块提取显著性的融合不同尺度的特征表征。三个公开数据集的实验结果表明,本文所提方法在去模糊性能、泛化能力和鲁棒性方面均优于现有方法。