计算机工程与应用 ›› 2023, Vol. 59 ›› Issue (17): 178-186.DOI: 10.3778/j.issn.1002-8331.2209-0050

• 图形图像处理 • 上一篇    下一篇

基于改进循环生成对抗网络的图像去噪研究

伏锦,黄山   

  1. 四川大学 电气工程学院,成都 610065
  • 出版日期:2023-09-01 发布日期:2023-09-01

Research on Image Denoising Based on Improved CycleGAN

FU Jin, HUANG Shan   

  1. College of Electrical Engineering, Sichuan University, Chengdu 610065, China
  • Online:2023-09-01 Published:2023-09-01

摘要: 图像去噪研究是图像处理中非常关键的一项技术。目前,常见的图像去噪方法大部分都需要成对的训练数据集,并且所生成的去噪图像大都会在颜色分布、边缘衔接等细节信息上和真实图像存在一定的差异,因此提出了一种基于改进的循环生成对抗网络的图像去噪方法。这种改进方法在原网络的基础上引入了多尺度判别器并提出了新的目标损失函数。其中,新的损失函数引入了像素损失和特征损失,还用Smooth L1范数损失代替了原网络中的L1范数损失。实验结果表明,提出的改进网络相较于原网络的性能有一定提升。和原网络相比,利用改进后的网络进行去噪的图片峰值信噪比从25.24?dB提高到29.02?dB,提高了15.0%;结构相似性指数从0.862提高到0.956,提高了10.9%。

关键词: 循环生成对抗网络, 损失函数, 多尺度判别器, 图像去噪, 图像处理, 深度学习

Abstract: Image denoising research is a very critical technology in image processing. At present, most of the common image denoising methods require paired training data sets, and the generated denoising images are mostly different from the real images in terms of color distribution, edge connection and other details. Therefore, an image denoising method based on improved cycle-consistent adversarial networks is proposed. A multi-scale discriminator is introduced and a new target loss function is proposed in the improved network. In the new target loss function, pixel loss and feature loss are introduced and the L1 norm loss in the original network is replaced with the Smooth L1 norm loss. The experimental results show that the improved network has a certain improvement in performance compared with the original network. Compared with the original network, the peak signal-to-noise ratio of the image denoised by the improved network is increased from 25.24 dB to 29.02 dB, an increase of 15.0%. The structural similarity index is increased from 0.862 to 0.956, an increase of 10.9%.

Key words: cycle-consistent adversarial networks, loss function, multi-scale discriminator, image denoising, image processing, deep learning