基于RDN和WGAN的图像超分辨率重建模型

作者：李易达; 马晓轩

来源：现代电子技术, 2021, 44(16): 79-84.

DOI：10.16652/j.issn.1004-373x.2021.16.016

• 图像重建模型
• RDN
• WGAN
• 图像处理
• 特征提取
• 模型设计

文中针对现有基于生成对抗网络的单图超分辨率重建模型训练不稳定,以及重建后的图像细节视觉效果不理想等问题,提出基于残差密集网络(RDN)和WGAN的图像超分辨率重建模型。模型使用残差密集块(RDB)作为生成器的基本结构单元,通过残差密集连接的卷积层提取丰富的细节特征,以提高重建后的图像质量;引入WGAN的思想,以EarthMover距离定义对抗损失以解决SRGAN模型的训练不稳定问题;采用Charbonnier损失函数衡量重建图像与高分辨率图像的相似程度,以提高重建图像的视觉效果。通过在Set5,Set14数据集上的实验表明,相比于双三次插值、SRCNN、RDN、SRGAN等SR重建模型,该文模型重建后图像的PSNR和SSIM评价指标更高,图像亮度信息、纹理细节及视觉效果均有提升。

• 单位
  北京建筑大学