CSA Computer Science and Application 2161-8801 Scientific Research Publishing 10.12677/csa.2024.145112 CSA-86907 csa2024145_41543231.pdf 信息通讯 基于深度残差生成对抗网络的超分辨率重建算法研究 Research on Super-Resolution Reconstruction Algorithms Based on Deep Residual Generative Adversarial Networks 刘胜 2 1 从哲 2 1 江苏理工学院计算机工程学院,江苏 常州 null 11 05 2024 14 05 33 47 © Copyright 2014 by authors and Scientific Research Publishing Inc. 2014 This work is licensed under the Creative Commons Attribution International License (CC BY). http://creativecommons.org/licenses/by/4.0/

图像超分辨重建是计算机视觉领域中的重要研究方向之一。本文主要针对图像超分辨率在重建过程中信息恢复不充分、特征提取不全面、高频细节不明显等问题,在SRGAN的基础上提出一种基于深度残差生成对抗网络的图像超分辨率算法。该算法实现了一种深度残差的结构(Depth-ResNet),即残差中的残差,来形成非常深的网络。该结构由几个具有长跳跃连接的残差组组成,每个残差组中也包含一些具有短跳跃连接的残差块。除此之外,Depth-ResNet允许通过多个跳跃连接绕过丰富的低频信息,使主网络专注于学习高频信息,并且随着Depth-ResNet的数量与深度的调整变化,图像重建效果将取得更好的准确性与视觉改进。此外,为了平衡Depth-ResNet在生成器上的卓越性能,本文在生成器的损失函数上采用了Charbonnier损失函数与对抗损失函数,并优化了判别器的结构。根据大量实验表明,重建的图像在清晰度、高频细节等方面都有一定的提高。 Image super-resolution reconstruction is one of the significant research directions within the field of computer vision. This paper particularly addresses issues such as insufficient information recovery, incomplete feature extraction, and indistinct high-frequency details during the image super-resolution process by proposing, on the basis of SRGAN, a novel deep residual generative adversarial network-based image super-resolution algorithm. The proposed algorithm implements a Depth-Residual Network (Depth-ResNet) architecture, characterized by residuals within residuals, to construct an extremely deep network. This architecture comprises several residual groups interconnected with long skip connections, where each residual group further consists of several residual blocks linked via short skip connections. Significantly, the Depth-ResNet design allows for multiple skip connections bypassing rich low-frequency information, enabling the main network to focus on learning high-frequency information effectively. As the number and depth of the Depth-ResNet modules are varied and adjusted, the reconstructed images exhibit improved accuracy and visually enhanced outcomes. Moreover, in order to balance the outstanding performance of the Depth-ResNet in the generator, this study adopts the Charbonnier loss function along with the adversarial loss function in the generator’s objective function, while also optimizing the structure of the discriminator. Extensive experimentation demonstrates that the reconstructed images exhibit noticeable improvements in terms of clarity and high-frequency detail preservation.

超分辨率重建,生成对抗网络,深度残差网络, Super-Resolution Reconstruction Generative Adversarial Networks Deep Residual Networks
摘要

图像超分辨重建是计算机视觉领域中的重要研究方向之一。本文主要针对图像超分辨率在重建过程中信息恢复不充分、特征提取不全面、高频细节不明显等问题,在SRGAN的基础上提出一种基于深度残差生成对抗网络的图像超分辨率算法。该算法实现了一种深度残差的结构(Depth-ResNet),即残差中的残差,来形成非常深的网络。该结构由几个具有长跳跃连接的残差组组成,每个残差组中也包含一些具有短跳跃连接的残差块。除此之外,Depth-ResNet允许通过多个跳跃连接绕过丰富的低频信息,使主网络专注于学习高频信息,并且随着Depth-ResNet的数量与深度的调整变化,图像重建效果将取得更好的准确性与视觉改进。此外,为了平衡Depth-ResNet在生成器上的卓越性能,本文在生成器的损失函数上采用了Charbonnier损失函数与对抗损失函数,并优化了判别器的结构。根据大量实验表明,重建的图像在清晰度、高频细节等方面都有一定的提高。

关键词

超分辨率重建,生成对抗网络,深度残差网络

Research on Super-Resolution Reconstruction Algorithms Based on Deep Residual Generative Adversarial Networks<sup> </sup>

Liusheng Wang, Congzhe You*

School of Computer Engineering, Jiangsu University of Technology, Changzhou Jiangsu

Received: Apr. 11th, 2024; accepted: May 9th, 2024; published: May 20th, 2024

ABSTRACT

Image super-resolution reconstruction is one of the significant research directions within the field of computer vision. This paper particularly addresses issues such as insufficient information recovery, incomplete feature extraction, and indistinct high-frequency details during the image super-resolution process by proposing, on the basis of SRGAN, a novel deep residual generative adversarial network-based image super-resolution algorithm. The proposed algorithm implements a Depth-Residual Network (Depth-ResNet) architecture, characterized by residuals within residuals, to construct an extremely deep network. This architecture comprises several residual groups interconnected with long skip connections, where each residual group further consists of several residual blocks linked via short skip connections. Significantly, the Depth-ResNet design allows for multiple skip connections bypassing rich low-frequency information, enabling the main network to focus on learning high-frequency information effectively. As the number and depth of the Depth-ResNet modules are varied and adjusted, the reconstructed images exhibit improved accuracy and visually enhanced outcomes. Moreover, in order to balance the outstanding performance of the Depth-ResNet in the generator, this study adopts the Charbonnier loss function along with the adversarial loss function in the generator’s objective function, while also optimizing the structure of the discriminator. Extensive experimentation demonstrates that the reconstructed images exhibit noticeable improvements in terms of clarity and high-frequency detail preservation.

Keywords:Super-Resolution Reconstruction, Generative Adversarial Networks, Deep Residual Networks

Copyright © 2024 by author(s) and beplay安卓登录

This work is licensed under the Creative Commons Attribution International License (CC BY 4.0).

http://creativecommons.org/licenses/by/4.0/

1. 引言

图像超分辨率重建(Image Super-resolution, SR)是指从低分辨率(LR)图像中重建为高分辨率(HR)图像的过程,是计算机视觉领域中重要的研究方向之一。该图像处理技术在现实生活中应用广泛,例如医学成像 [ 1 ] 、卫星遥感 [ 2 ] 、监控成像 [ 3 ] 与目标识别 [ 4 ] 等。除了提高图像感知质量外,图像超分辨率重建还有助于其他计算机视觉任务 [ 5 ] 的完成。根据前人的研究,当下主流的图像超分辨率方法包括基于预测的方法 [ 6 ] 、基于边缘的方法 [ 7 ] 、基于补丁的方法 [ 8 ] 、基于稀疏表示的方法 [ 9 ] ,以及基于统计的方法 [ 10 ] 等。

近年来随着深度学习的快速发展,基于深度学习的图像超分辨率模型得到了显著提升与完善。2014年,Dong等人 [ 11 ] 首次为图像超分辨率引入三层卷积神经网络(CNN),并提出了著名的SRCNN。此后,Kim等人在VDSR [ 12 ] 和DRCN [ 13 ] 中将网络深度增加到20层,相较于SRCNN的图像超分辨率效果取得了显著的提高。可见,模型的网络深度对许多计算机视觉任务起着不可或缺的作用,这一观点在He等人提出的残差网络(ResNet) [ 14 ] 中被证实。这种有效的残差学习策略随后被引入到许多其他基于卷积神经网络的图像超分辨率方法中。2017年,Lim等人 [ 15 ] 通过使用简化的残差块构建了较为宽泛的网络模型EDSR和深度较深的网络模型MDSR。EDSR与MDSR在性能上的显著提升,再次表明了模型的网络深度对图像超分辨率性能至关重要。同年,Ledig等人受生成对抗网络(Generative Adversarial Networks, GAN) [ 16 ] 的启发,将该算法思想应用于图像超分辨率重建,并提出了基于生成对抗网络的超分辨率图像重建(SRGAN) [ 17 ] 。SRGAN模型中同样包含了残差网络结构,但通过大量实验显示,简单地堆叠增加残差块的数量来构建更深层的网络,很难获得理想的效果与性能的提升。为了解决该问题,本文提出了一种深度残差的网络结构(Depth-ResNet),即残差中的残差,以形成较深的可训练网络。该网络结构中所包含的长短跳跃连接,使得图像中的低频信息可以被尽可能的绕过,从而一定程度上缓解了信息流动,更好地把握住图像中的高频细节,最终提高了图像超分辨率视觉效果。

2. 提出的方法 2.1. 方法概述

本节将描述所提出的模型架构。本文首先分析了近些年基于SRGAN的相关研究工作,并提出一种效果更好的残差网络架构Depth-ResNet,实现了残差中包含残差的效果,增强了网络对高频细节特征的提取能力。为了平衡在生成器中引入Depth-ResNet所带来的较深网络,本文在损失函数中引入了Charbonnier损失函数与对抗损失函数,并优化了判别器的网络结构,在一定程度上提高了图像的重建质量。

2.2. 残差块的改进

自从He等人在里程碑式的研究 [ 14 ] 中首次提出了残差网络(ResNet)这一概念以来,该架构已在众多计算机视觉任务中展现出卓越的性能表现。尽管Ledig等人随后在文献 [ 17 ] 中对原始ResNet架构进行了针对性优化,进而衍生出适用于图像超分辨问题的SRResNet模型,并在该领域取得了显著成效;然而,本文通过采用一种更为先进的残差网络构造方式,进一步提升了性能。

图1. ResNet、SRResNet、与Depth-ResNet残差块的比较

图1中比较了原始ResNet、SRResNet与本文所提出的Depth-ResNet的残差块结构。本文提出的Depth-ResNet残差块的构造参考了Nah等人 [ 18 ] 的做法,将网络内部的批量规一化层(BN)移除。鉴于批量归一化层对图像中的特征进行标准化操作,其内在地限制了网络对输入数据动态范围的适应性,故而移除该层可以释放网络的灵活性并有利于性能的提升。此外,LeakyReLU激活函数能够间接影响模型对输入数据分布的要求,在一定程度上可以降低对额外数据标准化步骤的需求。因此,本文选择将原始的ReLU激活函数替换为LeakyReLU激活函数,并引入Depth-ResNet残差块中,从而在图像特征处理中促进信息流动、改善深层网络的训练效率。

2.3. 实现深度残差的Depth-ResNet

本文所提出的Depth-ResNet实现了深度残差的效果,即残差网络中包含了若干残差网络。该网络结构的设计采用了多层级、多层次的残差组织方式,由一系列相互连接的长跳跃残差组构成,在每个残差组内部,又进一步整合了若干具有短跳跃连接的残差块单元。这一设计旨在通过多重连接路径实现图像特征的有效传递与整合。具体来说,Depth-ResNet网络架构的独特之处在于其能够利用长跳跃连接跨越多层网络深度,有效地保留并传播图像中的低频信息,从而使得主干网络无需重复处理这些基础特征,可以更专注于学习和捕获高频及细节丰富的信息内容。

图2. Depth-ResNet-1的网络架构

图3. Depth-ResNet-2的网络架构

图4. Depth-ResNet-3的网络架构

图2、图3、图4给出了关于Depth-ResNet的更多细节信息,在上述图中使用“Depth-ResNet-i”标识不同的网络架构,其中“i”表示网络中包含的残差块的数量,也可以认为是对网络深度的衡量。Depth-ResNet的最外层连接即为长跳跃连接,向内即为短跳跃连接。这种残差结构能够改善深层网络的训练效率与性能。

受前人工作成果 [ 17 ] 的启发,本文将Depth-ResNet中深度特征提取模块Depth-Block表示为 F ( x ) ,将Depth-ResNet-i中深度残差模块表示为 H i − 1 ( x ) ,其中 x 表示网络的输入,此处使用 x l 表示输入的低分辨率图像LR, x o u t 表示输出的重建后图像SR。则Depth-ResNet-1的输出可表示为:

x o u t 1 = F ( x l ) + x l (1)

进一步使用深度残差模块表示Depth-ResNet-2与Depth-ResNet-3的输出:

x o u t 2 = H 1 ( x l ) + x l = F ( F ( x l ) ) + F ( x l ) + x l (2)

x o u t 3 = H 2 ( x l ) + x l = F ( F ( F ( x l ) ) ) + F ( F ( x l ) ) + F ( x l ) + x l (3)

由上述(1)、(2)、(3)式可以归纳出Depth-ResNet-i的输出表达式:

x o u t i = H i − 1 ( x l ) + x l = F i ( F i − 1 ( ⋅ ⋅ ⋅ F 1 ( x l ) ⋅ ⋅ ⋅ ) ) + F i − 1 ( ⋅ ⋅ ⋅ F 1 ( x l ) ⋅ ⋅ ⋅ ) + ⋅ ⋅ ⋅ + F ( x l ) + x l (4)

由(4)式可知,随着Depth-ResNet深度的调整与扩展,深度特征提取模块Depth-Block对低分辨率图像 x l 的信息抽取次数不断增多,这一过程不仅加深了模型对图像结构的理解,更强化了对图像高频细节特征的学习与表达能力。这是因为每一层Depth-Block都会在前一层的基础上进一步细化特征图谱,更加关注那些能够揭示图像边缘、纹理等细微变化的高频成分。

此外,网络中的长跳跃连接跨越了多个层级,将浅层网络中的低频信息直接传递至深层网络中,使得在网络学习过程中,主干网络可以避免重复处理这些基础特征信息,从而集中精力去捕捉并保留更为复杂且具有区分度的高频特征。存在于残差组内部的短跳跃连接,确保了局部层级间的特征得以有效整合与传播,有助于优化梯度流,并在一定程度上缓解了深度神经网络在训练过程中梯度消失的问题。简而言之,Depth-ResNet通过深度残差的结构配置与层次间信息流动的机制,在保持模型深层表达能力的同时,有效地提升了图像重建任务的准确性与视觉表现力。

2.4. 基于Depth-ResNet的生成对抗网络

本文基于Depth-ResNet的生成对抗网络,由生成器(Generator)与判别器(Discriminator)组成。

生成器(图5)将低分辨率图像(LR)作为输入,依次通过一层卷积层、num个不同深度的Depth-ResNet,以及一个上采样模块与一层卷积层进行图像重建。根据实验条件,此处Depth-ResNet数量num取8、16、24、32,深度 i ≤ 10 。

判别器则是用来判断生成器生成的高分辨率图像(SR)与原始图像(HR)是否相似,并加以区分。本文提出的判别器模型(图6)由多个D-Block模块组成,每个D-Block由一层卷积层、批量归一化BN层与LeakyReLU激活函数构成,通过7个D-Block模块组合的方式对图像高频细节特征进行深度提取。因为批量归一化BN层在判别任务中效果较好,且能稳定网络的训练,所以本文在设计D-Block模块时选择保留BN层。判别器本质是一个二分类问题,而Sigmoid层主要功能是提供一种连续且可微的概率估计,并且具有平滑性与饱和性的特点,在解决二分类问题时,可以有效地处理复杂的决策边界。因此,本文在判别器的最后一层中选择使用Sigmoid层进行处理。以下该式可以表示判别器的二分类问题:

D ( x h , x s ) = α ⋅ ( 1 N ∑ i = 1 N | I ( x h ) i − I ( x s ) i | ) (5)

其中 α 表示Sigmoid激活函数、 N 表示图像的总像素数量, I ( x h ) i 与 I ( x s ) i 分别表示原始图像 x h 与生成器生成的高分辨率图像 x s 在第 i 个像素处的灰度值。上述(5)式的基本思想是衡量生成器生成的高分辨率图像 x s 与原始图像 x h 之间的差异,并进行二分类。

基于Depth-ResNet的生成对抗网络结构如下图所示。

图5. 生成器的网络结构

图6. 判别器的网络结构

2.5. 损失函数

本文算法的损失函数参考Lai [ 19 ] 等人的做法,在SRGAN生成器对抗损失函数的基础上引入抗噪性更强、鲁棒性更好、对图像边缘细节更敏感的Charbonnier损失函数,旨在网络训练过程中加强对图像高频细节特征的提取。Charbonnier损失函数如下:

L c h a r b o n n i e r ( x h , x s ) = 1 N ∑ i = 1 N α ( x h − x s ) (6)

上式中 x h 为原始图像HR, x s 为生成器生成的高分辨率图像SR, α ( x ) = x 2 + σ 2 ( σ = 1 × 10 − 8 ) 为加入的惩罚项。SRGAN中的对抗损失旨在网络训练过程中,通过判别器的输出对生成器模型的参数进行优化,从而实现两个网络间的动态博弈过程。此外,对抗损失还鼓励生成器产生更多自然、随机且丰富的细节特征,并且允许模型学习从低分辨率图像到高分辨率图像的复杂非线性映射关系,因此本文选择保留SRGAN中的对抗损失,其表达式如下:

L a d v e r ( x l ) = − ∑ i = 1 N log D ( G ( x l ) ) (7)

上式中 x l 为输入网络的低分辨率图像LR,则 G ( x l ) 表示的是生成器生成的高分辨率图像SR, D ( G ( x l ) ) 表示的是判别器将生成器生成的图像SR判别为原始图像HR的概率。综合以上损失函数,可以得到生成器的总损失 L t o t a l :

L t o t a l = λ 1 L c h a r b o n n i e r + λ 2 L a d v e r (8)

上式中 λ 1 、 λ 2 为所选取的权重系数,本文取 λ 1 = λ 2 = 1。

3. 实验结果与分析

本文实验训练使用Ubuntu20.04操作系统、Python3.10、Pytorch2.0.1、CUDA11.8,CPU型号为Intel(R) Xeon (R) Platinum 8260 CPU @2.30GHz,GPU型号为NVIDIA GeForce RTX 3090,显存为24 G。本实验的训练数据集采用DIV2K数据集中的800张训练图像,测试集使用基准数据集Set5、Urban100、BSD100、Sun-Hays80、Managa109。实验训练过程中使用Adam优化器,参数设置如下: β 1 = 0.9, β 2 = 0.999, ε = 1 × 10 − 8 ,学习率lr为 1 × 10 − 4 。本实验为了探究Depth-ResNet深度与数量对于模型性能的影响,并得出最优的Depth-ResNet深度与数量组合,将Depth-ResNet的深度划分为1至10,每个深度归为一组,对应Depth-ResNet数量为8、16、24、32。本文在Set5基准数据集上对上述组合进行4倍放大的测试,由于实验条件的限制,模型的参数量应小于 1 × 10 7 。为了验证最优Depth-ResNet深度与数量组合的有效性,本文选取了传统的超分方法双三次插值法Bicubic与基于学习的方法EDSR [ 15 ] 、MDSR [ 15 ] 、SRGAN [ 17 ] 来进行比较实验。

3.1. 评价指标

本文使用峰值信噪比(Peak Signal-to-Noise Ratio, PSNR)与结构相似性指数(Structural Similarity Index, SSIM)来对重建图像质量进行评估。其中PSNR用于衡量两幅图像之间的最大可能误差,PSNR的值越大,表示两幅图像之间的差异越小,图像重建质量越好。PSNR的计算公式如下:

R P S N R = 10 log 10 ( x max 2 1 W × H ∑ i = 1 W ∑ j = 1 H ( x h ( i , j ) − x s ( i , j ) ) 2 ) (9)

上式中 x max 表示图像中可能的最大像素值,对于彩色图像来说此值取 2 8 − 1 ,即255。式中 x h 与 x s 分别表示原始图像HR与生成器生成的高分辨率图像SR,W与H分别表示图像的宽度与高度, ( x h ( i , j ) − x s ( i , j ) ) 则表示原始图像HR与生成器生成的高分辨率图像SR对应像素点之间的差异。

由于PSNR只与像素级MSE有关,并且只关注对应像素之间的差异,因此若要更好地衡量重建图像的质量,还需引入一种更复杂的图像质量评估指标SSIM。SSIM不仅考虑了像素值之间的差异,还考虑了图像的结构信息,其计算公式如下:

S S S I M = ( 2 ⋅ μ x h ⋅ μ x s + c 1 μ x h 2 + μ x s 2 + c 1 ) α ( 2 ⋅ σ x h x s + c 2 σ x h 2 + σ x s 2 + c 2 ) β ( σ x h x s 2 + c 3 σ x h 2 + σ x s 2 + c 3 ) γ (10)

其中 μ x h 与 μ x s 分别表示原始图像HR与生成器生成的高分辨率图像SR的均值, σ x h x s 表示两幅图像的协方差, σ x h 2 与 σ x s 2 分别表示两幅图像的方差。为了避免分母为零,故在(10)式中加入 c 1 、 c 2 、 c 3 三个小常数。式中的 α 、 β 、 γ 为所选取的权重系数,通常取值为1。SSIM值的范围在0至1之间,其值越接近于1,表示两幅图像的结构相似度越高,图像重建质量越好。

3.2. 实验量化结果分析

图7、图8是由表1数据绘制的折线图,用以反映Depth-ResNet不同深度与数量组合对模型性能的影响。在表1中,我们使用黑体标注出不同深度Depth-ResNet中最佳实验数据,并以下划线标识出全局最优实验效果的数据。

Comparison of depth and quantity experimental results of Depth-ResNet (Depth-ResNet: Depths up to 10, parameter count less than or equal to 1 × 107
Depth-ResNet深度 Depth-ResNet数量 Depth-Block数量 参数量 PSNR (×4 Set5) SSIM (×4 Set5)
1 8 8 889,819 31.446 0.736
16 16 1,480,667 31.447 0.733
24 24 2,071,515 31.434 0.733
32 32 2,662,363 31.425 0.731
2 8 16 1,480,667 31.438 0.733
16 32 2,662,363 31.444 0.735
24 48 3,844,059 31.422 0.733
32 64 5,025,755 31.493 0.736
3 8 24 2,071,515 31.467 0.737
16 48 3,844,059 31.392 0.730
24 72 5,616,603 31.456 0.734
32 96 7,389,147 31.448 0.734
4 8 32 2,662,363 31.439 0.735
16 64 5,025,755 31.450 0.733
24 96 7,389,147 31.463 0.735
32 128 9,752,539 31.425 0.734
5 8 40 3,253,211 31.410 0.733
16 80 6,207,451 31.448 0.736
24 120 9,161,691 31.418 0.733
32 - - - -
6 8 48 3,844,059 31.424 0.735
16 96 7,389,147 31.475 0.735
24 - - - -
32 - - - -
7 8 56 4,434,907 31.460 0.737
16 112 8,570,843 31.443 0.736
24 - - - -
32 - - - -
8 8 64 5,025,755 31.450 0.735
16 128 9,752,539 31.437 0.735
24 - - - -
32 - - - -
9 8 72 5,616,603 31.466 0.737
16 - - - -
24 - - - -
32 - - - -
10 8 80 6,207,451 31.434 0.733
16 - - - -
24 - - - -
32 - - - -

表1. Depth-ResNet的深度与数量实验结果对比(Depth-ResNet深度 ≤ 10,参数量 ≤ 1 × 107)

本文在2.3节中,用理论的方式推出随着Depth-ResNet深度的调整与扩展,网络中的Depth-Block模块能够更频繁地抽取图像的特征,在高频细节方面有更强的学习与表现能力,所对应重建图像的峰值信噪(PSNR)比与结构相似性指数(SSIM)将会逐渐提高。但是由表1实验数据对应的折线图显示,不同数量的Depth-ResNet组合的性能随着Depth-ResNet深度的增加,并非呈现明显的递增趋势,而是呈现出一种规律性的上下浮动。即增加Depth-ResNet网络的复杂性并不一定会带来理想的性能提升,反而有可能出现过拟合现象导致模型性能的下滑。因此,应当合理权衡网络的深度与宽度,选择合适的Depth-ResNet深度与数量的组合,以达到最优的模型性能。

根据图表中数据显示,对于复杂度较低、Depth-ResNet数量组合为8与16的网络来说,选取奇数深度,模型将会有较好的性能表现;而对于复杂度较高、Depth-ResNet数量组合为24与32的网络而言,选取偶数深度,模型的性能将会表现较好。

由图8折线图显示,当网络中Depth-Block的数量选取在60至80之间,过拟合现象将会对模型的影响较小,性能表现较为优异。

图7. Depth-ResNet不同深度对应的模型性能对比

图8. Depth-Block不同数量对应的模型性能对比

图9为表1中不同深度Depth-ResNet的最佳实验数据组别所绘制的散点图,其中使用红色加粗点标注的是全局最优实验数据组别。从图9中可以发现Depth-ResNet深度为2、数量为32的网络模型参数量在 5 × 10 6 左右,这个数值恰好接近于散点图中所有不同组别参数量分布的平均水平,表明该模型的参数规模处在适宜的区间之内。综合上文所述,在本文的实验条件下,最优的Depth-ResNet深度与数量组合如下:Depth-ResNet深度为2、数量为32,Depth-Block数量为64的网络模型。

图9. Depth-ResNet不同“深度–数量”组合的性能与参数量对比

为了验证最优Depth-ResNet深度与数量组合的有效性,本文选取了传统的超分方法双三次插值法Bicubic与基于学习的方法EDSR [ 15 ] 、MDSR [ 15 ] 、SRGAN [ 17 ] 来进行比较实验。并在基准数据集Set5、Urban100、BSD100、Sun-Hays80与Manga109上进行尺度因子×2、×3、×4的对比实验。

Objective evaluation metric comparison at scale factors ×2, ×3, and ×4 on benchmark dataset
方法 倍率 Set5 Urban100 BSD100 Sun-Hays80 Manga109
PSNR SSIM PSNR SSIM PSNR SSIM PSNR SSIM PSNR SSIM
Bicubic ×2 33.191 0.886 31.274 0.799 32.197 0.807 33.664 0.879 33.987 0.901
EDSR 33.745 0.875 31.369 0.780 32.785 0.827 34.722 0.899+ 33.589 0.772
MDSR 34.185 0.893 31.514 0.792* 32.909 0.834* 34.873 0.905* 34.142 0.805
SRGAN 33.863 0.878 31.391 0.780 32.805 0.828 34.752 0.899+ 33.709 0.777
ours 34.196 0.890 31.523 0.792* 32.918 0.834* 34.886 0.905* 34.175 0.798
Bicubic ×3 31.785 0.793 30.574 0.645 31.240 0.661 32.352 0.762 32.574 0.771
EDSR 32.672 0.827 31.082 0.700 32.080 0.728* 33.572 0.826 32.388 0.679
MDSR 32.917 0.839 31.220 0.711 32.169 0.734 33.661 0.832 32.674 0.705
SRGAN 32.576 0.824 31.060 0.699 32.048 0.728* 33.527 0.825 32.278 0.680
ours 33.047 0.842 31.289 0.716 32.227 0.738 33.732 0.835 32.878 0.718
Bicubic ×4 30.855 0.677 29.991 0.509 30.610 0.555 31.540 0.658 31.962 0.693
EDSR 31.182 0.718 30.131 0.573 31.062 0.624 32.226 0.727 31.466 0.613
MDSR 31.379 0.732 30.216 0.582 31.140 0.631 32.309 0.734 31.736 0.637
SRGAN 31.119 0.714 30.103 0.570 31.042 0.621 32.201 0.725 31.397 0.606
ours 31.493 0.736 30.252 0.585 31.186 0.634 32.363 0.737 31.855 0.638

表2. 基准数据集尺度因子×2、×3、×4客观评价指标对比

在表2中,我们分别使用黑体和下划线来标识出每组实验中的最优结果与次优结果,并对数值相同的结果采取了相同的记号标注在对应数据的右上角以示区分。

从整体视角审视,最佳的Depth-ResNet深度与数量组合模型,在面对不同尺度因子下的五个基准数据集测试时,其性能均位列各组实验的最优结果与次优结果之中,并且最优结果在总实验中占据了较大比例。

从每组实验结果来看,最佳的Depth-ResNet深度与数量组合模型在3倍与4倍的重建任务中表现较为优异,最优结果占所有实验的绝大多数,说明Depth-ResNet在处理复杂且难度较高的图像重建任务时,对图像细节与高频特征的提取具有出色的性能表现。

而在2倍重建任务中,可以发现小部分测试集中最佳Depth-ResNet组合模型处理结果略逊色于MDSR,甚至在Urban100测试集上比Bicubic方法的SSIM值还落后0.007。据实验分析,这是由于图像重建倍率较小时,输入的低分辨率图像本身已经包含了明显的特征信息,使用MDSR就足够充分提取关键信息,少部分重建任务使用Bicubic即可完成。对表2中的数据继续分析,发现在Manga109基准数据集上,传统的插值方法Bicubic在各个尺度因子下重建效果均较好,而包括最佳Depth-ResNet组合模型在内的基于学习的方法重建效果却表现并不理想。回顾模型的训练过程,发现基于学习的各类模型都是在DIV2K数据集上进行训练的,DIV2K数据集中的图像包含的主要是各种自然场景,涵盖的是日常风景、人物与建筑等主题场景。而本实验中使用的Manga109数据集包含的则是动漫风格的图像,与日常生活的风景场景相比,其艺术风格迥异,色彩丰富且线条特征明显。此类图像往往具有独特的纹理与绘画风格,不同于基于现实生活图像训练的模型所擅长处理的内容。因此,基于学习的各类方法在此基准数据集上的表现较差。相反,传统的插值方法Bicubic是通过局部像素邻域信息进行插值,不易受数据集种类的影响,故其在Manga109数据集上重建效果较好。

根据表2实验数据,需要注意的是在Manga109基准数据集上,最佳Depth-ResNet组合模型虽然不擅长处理此类场景的图像,但是其重建数据依然位列于各组实验的最优结果与次优结果中,说明Depth-ResNet可以很好地把握图像中的高频细节,并提高图像超分辨率效果。

3.3. 纹理细节对比分析

图10、图11、图12呈现了最佳Depth-ResNet深度与数量组合模型与其他重构方法在各个基准数据集上的性能比较结果,其中最优模型的峰值信噪比(PSNR)和结构相似性指数(SSIM)以黑体标注显示。

图10. 在Set5基准数据集上×4超分辨率的定性比较

图11. 在Urban100、BSD100、Sun-Hays80基准数据集上×4超分辨率的定性比较

图12. 在Manga109基准数据集上×4超分辨率的定性比较

通过观察可以发现,最佳Depth-ResNet组合模型在重建图像时不仅取得了最高的PSNR与SSIM值,而且其重建图像在高频细节表现上更为突出,纹理层次更为清晰,从而呈现出更优的视觉效果。

4. 结束语

本文提出的基于深度残差生成对抗网络的图像超分辨率算法,实现了一种深度残差的结构(Depth-ResNet),即在残差中嵌套残差,来形成非常深的网络。Depth-ResNet通过多个跳跃连接绕过了一定数量的低频信息,从而使得主网络专注于学习图像中的高频信息。此外,为了平衡Depth-ResNet在生成器上的性能,本文在生成器损失函数上采用了Charbonnier损失函数与对抗损失函数,并优化了判别器的结构。随着Depth-ResNet数量与深度的调整变化,图像重建效果将呈现出一种规律性的上下浮动,本文实验发现:对于复杂度较低的网络来说,选取奇数深度的Depth-ResNet,模型将会有更好的性能表现;而对于复杂度较高的网络而言,选取偶数深度,模型的性能将会有一定程度的提高。并且当网络中Depth-Block的数量选取在60至80之间,过拟合现象将会对模型的影响较小。除此之外,本文通过大量实验得出了最优的Depth-ResNet深度与数量组合如下:Depth-ResNet深度为2、数量为32,Depth-Block数量为64的网络模型。实验结果显示,该模型相较于传统的插值方法Bicubic与基于学习的方法EDSR [ 15 ] 、MDSR [ 15 ] 、SRGAN [ 17 ] 在不同放大倍率、不同基准数据集下都具有不小的优势。并且该最佳Depth-ResNet组合模型的参数规模处于实验中的平均水平,分布于合理区间内,具有一定的应用价值。但本文在实验中发现,增加Depth-ResNet网络的复杂度与模型性能的提升并不是简单的线性关系,反而有可能出现过拟合现象导致模型性能的下滑。因此,解决该问题,并平衡网络复杂度与模型性能的关系可作为下一步的研究方向。

基金项目

本研究得到了江苏省高等学校大学生创新创业训练计划项目“基于生成对抗网络的超分辨率图像重构研究”(202311463057Y)的支持。

文章引用

王刘胜,由从哲. 基于深度残差生成对抗网络的超分辨率重建算法研究Research on Super-Resolution Reconstruction Algorithms Based on Deep Residual Generative Adversarial Networks[J]. 计算机科学与应用, 2024, 14(05): 33-47. https://doi.org/10.12677/csa.2024.145112

参考文献 References Greenspn, H. (2008) Super-Resolution in Medical Imaging. The Computer Journal, 52, 43-63.
https://doi.org/10.1093/comjnl/bxm075
Su, J.M. and Yang, L.X. (2019) Super Resolution of Single Frame Remote Sensing Image Based on Generative Adversarial Nets. Computer Engineering and Applications, 55, 202-207. Zou, W.W. and Yuen, P.C. (2012) Very Low Resolution Face Recognition Problem. IEEE Transactions on Image Processing, 21, 327-340.
https://doi.org/10.1109/TIP.2011.2162423
Sajjadi, M.S., Schölkopf, B. and Hirsch, M. (2017) EnhanceNet: Single Image Super-Resolution through Automated Texture Synthesis. 2017 IEEE International Conference on Computer Vision (ICCV), Venice, 22-29 October 2017, 4501-4510.
https://doi.org/10.1109/ICCV.2017.481
Dai, D., Wang, Y., Chen, Y. and Van Gool, L. (2016) Is Image Super-Resolution Helpful for Other Vision Tasks? 2016 IEEE Winter Conference on Applications of Computer Vision (WACV), Lake Placid, 7-10 March 2016, 1-9.
https://doi.org/10.1109/WACV.2016.7477613
Keys, R. (1981) Cubic Convolution Interpolation for Digital Image Processing. IEEE Transactions on Acoustics, Speech, and Signal Processing, 29, 1153-1160.
https://doi.org/10.1109/TASSP.1981.1163711
Freedman, G. and Fattal, R. (2011) Image and Video Upscaling from Local Self-Examples. ACM Transactions on Graphics, 30, Article No. 12.
https://doi.org/10.1145/1944846.1944852
Freeman, W.T., Jones, T.R. and Pasztor, E.C. (2002) Example-Based Super-Resolution. IEEE Computer Graphics and Applications, 22, 56-65.
https://doi.org/10.1109/38.988747
Yang, J., Wright, J., Huang, T.S. and Ma, Y. (2010) Image Super-Resolution via Sparse Representation. IEEE Transactions on Image Processing, 19, 2861-2873.
https://doi.org/10.1109/TIP.2010.2050625
Kim, K.I. and Kwon, Y. (2010) Single-Image Super-Resolution Using Sparse Regression and Natural Image Prior. IEEE Transactions on Pattern Analysis and Machine Intelligence, 32, 1127-1133.
https://doi.org/10.1109/TPAMI.2010.25
Dong, C., Loy, C.C., He, K. and Tang, X. (2014) Learning a Deep Convolutional Network for Image Super-Resolution. 13th European Conference Computer Vision-ECCV 2014, Zurich, 6-12 September 2014, 184-199.
https://doi.org/10.1007/978-3-319-10593-2_13
Kim, J., Kwon Lee, J. and Mu Lee, K. (2016) Accurate Image Super-Resolution Using Very Deep Convolutional Networks. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 1646-1654.
https://doi.org/10.1109/CVPR.2016.182
Kim, J., Kwon Lee, J. and Mu Lee, K. (2016) Deeply-Recursive Convolutional Network for Image Super-Resolution. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 1637-1645.
https://doi.org/10.1109/CVPR.2016.181
He, K., Zhang, X., Ren, S. and Sun, J. (2016) Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 770-778.
https://doi.org/10.1109/CVPR.2016.90
Lim, B., Son, S., Kim, H., Nah, S. and Lee, K.M. (2017) Enhanced Deep Residual Networks for Single Image Super-Resolution. 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Honolulu, 21-26 July 2017, 1132-1140.
https://doi.org/10.1109/CVPRW.2017.151
Goodfellow, I.J., Mirza, M., et al. (2014) Generative Adversarial Nets. Advances in Neural Information Processing Systems, Vol. 27, 2672-2680. Ledig, C., Theis, L., Huszár, F., et al. (2017) Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, 21-26 July 2017, 4681-4690.
https://doi.org/10.1109/CVPR.2017.19
Nah, S., Kim, T.H. and Lee, K.M. (2016) Deep Multi-Scale Convolutional Neural Network for Dynamic Scene Deblurring. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, 21-26 July 2017, 257-265.
https://doi.org/10.1109/CVPR.2017.35
Lai, W.S., Huang, J.B., Ahuja, N., et al. (2017) Deep Laplacian Pyramid Networks for Fast and Accurate Super-Resolution. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, 21-26 July 2017, 624-632.
https://doi.org/10.1109/CVPR.2017.618
Baidu
map