《计算机应用》唯一官方网站 ›› 0, Vol. ›› Issue (): 0-0.DOI: 10.11772/j.issn.1001-9081.2023081131
• • 下一篇
丁宇伟1,石洪波2,李杰3,梁敏4
摘要: 摘 要: 当前基于Transformer的图像去噪算法侧重于捕获图像的全局特征,而忽视了局部特征对于恢复图像细节信息的关键作用。针对该问题,提出一种基于局部和全局特征解耦的图像去噪算法,它包含两个基于混合Transformer模块(HTB)的多尺度分支和一个基于卷积神经网络(CNN)的单尺度分支,旨在将HTB强大的全局建模能力与CNN的局部建模优势有机结合,生成上下文信息丰富且空间细节准确的输出。HTB采用自注意力机制自适应地对空间和通道维度的依赖关系进行建模,以激活范围更广的输入像素进行重建。鉴于不同分支间可能存在的信息冲突,设计了特征传递模块,通过跨分支传递全局特征并抑制低频信息,从而确保各分支间的协同作用。在真实世界图像数据集DND上,与基于Transformer的去噪算法(Uformer)相比,所提算法的峰值信噪比(PSNR)提高了0.14dB。在合成图像数据集Urban100上,与多阶段去噪算法(MSPNet)相比,所提算法的平均PSNR提高了0.41dB。实验结果表明,所提算法能够有效去除图像噪声,并重建出更精细的纹理细节。
中图分类号: