arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.09000cs.CVcs.LG

用于椒盐图像去噪的残差Transformer融合网络

Residual Transformer Fusion Network for Salt and Pepper Image Denoising

Universitas Sebelas Maret(塞贝拉斯·马雷特大学)

Bintang Pradana Erlangga Putra, Heri Prasetyo, Esti Suryani

AI总结:

针对部分图像去噪方法需要噪声先验知识的问题,提出结合CvT与ResNet的RTF-Net架构,在多噪声水平测试中除30%噪声的Pepper图像外均优于对比方法。

摘要:

卷积神经网络(CNN)已广泛应用于非结构化数据集,图像去噪便是其中之一。图像去噪是一个噪声图像重建过程,旨在通过各种策略减少噪声图像中出现的额外噪声。图像去噪存在一个问题,即部分图像去噪方法需要一些关于噪声信息的先验知识。为解决这一问题,本文使用了卷积视觉Transformer(CvT)与残差网络(ResNet)相结合的架构,称为残差Transformer融合网络(RTF-Net)。总体而言,该架构中的处理过程可分为两部分:噪声抑制网络(NSN)和结构增强网络(SEN)。残差块用于噪声抑制网络,用于学习图像中的噪声图;而CvT用于结构增强网络,用于学习需要添加到经噪声抑制网络处理后的图像中的细节。该模型使用DIV2K训练集进行训练,使用DIV2K验证集进行验证。训练完成后,使用Lena、Bridge、Pepper和BSD300图像对模型进行测试,噪声水平分别为30%、50%和70%,并将峰值信噪比(PSNR)结果与DBA、NASNLM、PARIGI、NLSF、NLSF-MLP和NLSF-CNN方法进行比较。测试结果表明,除噪声水平为30%的Pepper图像外,所提方法在所有情况下均表现更优;在该例外情况中,NLSF-CNN以32.99 dB的PSNR值更优,而所提方法的PSNR值为31.70 dB。

英文摘要:

Convolutional Neural Network (CNN) has been widely used in unstructured datasets, one of which is image denoising. Image denoising is a noisy image reconstruction process that aims to reduce additional noise that occurs from the noisy image with various strategies. Image denoising has a problem, namely that some image denoising methods require some prior knowledge of information about noise. To overcome this problem, a combined architecture of Convolutional Vision Transformer (CvT) and Residual Networks (ResNet) is used which is called the Residual Transformer Fusion Network (RTF-Net). In general, the process in this architecture can be divided into two parts, Noise Suppression Network (NSN) and Structure Enhancement Network (SEN). Residual Block is used in the Noise Suppression Network and is used to learn the noise map in the image, while the CvT is used in the Structure Enhancement Network and is used to learn the details that need to be added to the image processed by the Noise Suppression Network. The model was trained using the DIV2K Training Set dataset, and validation using the DIV2K Validation Set. After doing the training, the model was tested using Lena, Bridge, Pepper, and BSD300 images with noise levels ranging from 30%, 50%, and 70% and the PSNR results were compared with the DBA, NASNLM, PARIGI, NLSF, NLSF-MLP and NLSF-CNN methods. The test results show that the proposed method is superior in all cases except for Pepper's image with a noise level of 30%, where NLSF-CNN is superior with a PSNR value of 32.99 dB, while the proposed method gets a PSNR value of 31.70 dB.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑