基于双注意力卷积及Transformer融合的非均匀去雾算法

针对现有大部分去雾算法中对不同雾霾区域关注不足以及浓雾区域细节信息恢复不理想的问题,提出了一种结合卷积神经网络和Transformer模块的非均匀去雾算法。首先,为了更好地关注浓雾区域,在浅层特征提取阶段构建了并联双注意力卷积网络,分别从像素和通道的角度给图像分配不同的权重;其次,在深层特征提取中,引入了Transformer模块进行全局非均匀雾霾区域特征提取,既能有效捕捉特征之间的长距离依赖关系,又避免了普通卷积扩大感受野导致细节信息丢失的问题;最后,设计了多特征融合重建网络,能够自适应地融合浅层和深层特征,从而重构清晰图像。在公共数据集和自建非均匀雾霾数据集上进行了大量实验,结果表明,所提算法在视觉效果和客观评价指标上均优于其他对比算法。