计算机技术与发展2024,Vol.34Issue(4) :48-54.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0008

基于U形多尺度注意力方法的真实图像去噪

Real-world Image Denoising Based on U-shaped Multi-scale Attention Method

王新武 陈春雨
计算机技术与发展2024,Vol.34Issue(4) :48-54.DOI:10.20165/j.cnki.ISSN1673-629X.2024.0008

基于U形多尺度注意力方法的真实图像去噪

Real-world Image Denoising Based on U-shaped Multi-scale Attention Method

王新武 1陈春雨1
扫码查看

作者信息

  • 1. 哈尔滨工程大学 信息与通信工程学院,黑龙江 哈尔滨 150001
  • 折叠

摘要

针对真实世界图像去噪算法存在对上下文信息和全局信息利用不足导致的去噪效果不佳问题,提出一种U形金字塔注意力网络(UPCA).U形结构由多尺度特征模块与长距离通道注意力模块融合形成的金字塔注意力模块组成,U形结构通过拼接操作可以将每一层的输出特征图融合,减少卷积过程以及下采样过程中图像细节特征的丢失.多尺度特征金字塔模块可以更好地利用上下文信息从而更好地恢复出干净的图像,而建立长距离依赖的通道注意力模块可以更好地利用全局信息,提高网络的去噪效果.同时在损失函数部分加入噪声项来加快训练时收敛的速度以及提高去噪效果.UPCA网络在数据集SIDD和DND进行对比实验,验证了 UPCA网络的可行性和先进性,同时与同样使用通道注意力的RIDNet相比UPCA网络的PSNR/SSIM指标提升了 0.81 dB/0.044,去噪后的效果图直观表现也更好,而且同等参数下训练所需的算力更小.

Abstract

To address the issue of subpar denoising results in existing algorithms for real-world image denoising,we propose an innovative solution called the U-Shape Pyramid Channel Attention(UPCA).The U-shape structure comprises a fusion of multi-scale feature modules and long-range channel attention modules,forming a pyramid attention module.Through concatenation operations,the U-shape structure allows for the fusion of output feature maps from each layer,minimizing the loss of fine-grained image details during the convolution and downsampling processes.The multi-scale feature pyramid module effectively leverages contextual information to restore clean images,while the long-range channel attention module establishes dependencies on global information,thereby enhancing the denoising performance of the network.Additionally,we introduce a noise term in the loss function to expedite convergence during training and improve denoising efficiency.Experimental comparisons on the SIDD and DND datasets demonstrate the feasibility and su-periority of the UPCA.Compared to RIDNet which also utilizes channel attention,UPCA achieves a remarkable improvement of 0.81 dB/0.044 in terms of PSNR/SSIM metrics.The visually enhanced denoised images produced by UPCA are superior,and it requires less computational power for training with the same set of parameters.

关键词

图像去噪/计算机视觉/真实噪声/多尺度特征/长距离通道注意力

Key words

image denoising/computer vision/real noise/multi-scale features/long-range channel attention

引用本文复制引用

基金项目

国家自然科学基金(61871142)

中央高校基本科研业务费专项(3072020CFT0803)

出版年

2024
计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子:0.621
ISSN:1673-629X
参考文献量19
段落导航相关论文