基于集成学习的双分支混合注意力块去雾算法

针对现有深度学习方法在处理真实雾图时会导致恢复图像颜色失真,去雾不彻底等问题,提出了一种基于集成学习的双分支混合注意力块去雾网络,该网络是由全局特征提取分支和高频纹理特征提取分支以及基于集成学习融合尾部组成。全局特征提取子网使用Res2Net作为预训练模型初始化权重,提高全局特征提取鲁棒性。高频纹理特征提取子网使用U-Net为骨干网络结合了混合注意力块,跳跃连接以及软重建,使该分支更能关注到当前数据集高频纹理特征,最终融合两分支并映射成清晰图像。该算法与最先进方法比较,在RESIDE合成数据集取得了良好的视觉效果,并在NTIRE2020和NTIRE2021数据集上客观指标较最新DehazeFormer去雾模型PSNR提升了12.09 dB、SSIM提升了40.7%。

资料为PDF文档格式.

本文档关键词:算法,注意力,分支,混合,学习