CN111476717A - 基于自注意力生成对抗网络的人脸图像超分辨重建方法 - Google Patents
基于自注意力生成对抗网络的人脸图像超分辨重建方法 Download PDFInfo
- Publication number
- CN111476717A CN111476717A CN202010262877.3A CN202010262877A CN111476717A CN 111476717 A CN111476717 A CN 111476717A CN 202010262877 A CN202010262877 A CN 202010262877A CN 111476717 A CN111476717 A CN 111476717A
- Authority
- CN
- China
- Prior art keywords
- layer
- network
- convolution
- generator
- resolution
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Granted
Links
- 238000000034 method Methods 0.000 title claims abstract description 71
- 238000012549 training Methods 0.000 claims abstract description 63
- 230000003042 antagnostic effect Effects 0.000 claims abstract description 21
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 claims description 133
- 230000006870 function Effects 0.000 claims description 120
- 230000004913 activation Effects 0.000 claims description 66
- 238000011176 pooling Methods 0.000 claims description 28
- 238000010606 normalization Methods 0.000 claims description 21
- 230000008569 process Effects 0.000 claims description 15
- ORILYTVJVMAKLC-UHFFFAOYSA-N Adamantane Natural products C1C(C2)CC3CC1CC2C3 ORILYTVJVMAKLC-UHFFFAOYSA-N 0.000 claims description 12
- 238000006731 degradation reaction Methods 0.000 claims description 8
- 238000011478 gradient descent method Methods 0.000 claims description 6
- 230000001902 propagating effect Effects 0.000 claims description 6
- 238000000605 extraction Methods 0.000 claims description 4
- 230000001815 facial effect Effects 0.000 claims description 3
- 239000005557 antagonist Substances 0.000 claims description 2
- 230000000694 effects Effects 0.000 abstract description 8
- 238000005070 sampling Methods 0.000 abstract description 5
- 238000004088 simulation Methods 0.000 description 15
- 238000010586 diagram Methods 0.000 description 12
- 238000005516 engineering process Methods 0.000 description 5
- 238000012360 testing method Methods 0.000 description 4
- 230000007547 defect Effects 0.000 description 3
- 238000013528 artificial neural network Methods 0.000 description 2
- 230000005540 biological transmission Effects 0.000 description 2
- 238000013461 design Methods 0.000 description 2
- 238000013507 mapping Methods 0.000 description 2
- 230000000007 visual effect Effects 0.000 description 2
- 238000004458 analytical method Methods 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 238000004891 communication Methods 0.000 description 1
- 238000010276 construction Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 239000011159 matrix material Substances 0.000 description 1
- 230000007246 mechanism Effects 0.000 description 1
- 238000012544 monitoring process Methods 0.000 description 1
- 238000012545 processing Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T3/00—Geometric image transformations in the plane of the image
- G06T3/40—Scaling of whole images or parts thereof, e.g. expanding or contracting
- G06T3/4053—Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T3/00—Geometric image transformations in the plane of the image
- G06T3/40—Scaling of whole images or parts thereof, e.g. expanding or contracting
- G06T3/4046—Scaling of whole images or parts thereof, e.g. expanding or contracting using neural networks
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T10/00—Road transport of goods or passengers
- Y02T10/10—Internal combustion engine [ICE] based vehicles
- Y02T10/40—Engine management systems
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- Biomedical Technology (AREA)
- Life Sciences & Earth Sciences (AREA)
- Health & Medical Sciences (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Image Analysis (AREA)
Abstract
本发明提出了一种基于自注意力生成对抗网络的人脸图像超分辨重建方法,解决了现有技术只能针对简单下采样所得低分辨图像进行重建,而对于真实场景中的低分辨图像重建效果差的问题。实现步骤为:构建两个生成器网络和判别器网络;构建并初始化两个自注意力生成对抗网络;生成初始人脸图像训练集;训练两个自注意力生成对抗网络;对低分辨人脸图像进行超分辨率重建。本发明在网络中加入了自注意力层,引入图像全局整体特征,在损失函数中加入了上下文损失,增强了生成图像的纹理细节信息。
Description
技术领域
本发明属于图像处理技术领域,更进一步涉及图像超分辨重建技术领域中的一种基于自注意力生成对抗网络的人脸图像超分辨重建方法。本发明可用于从真实场景中的低分辨人脸图像重建出对应的高分辨人脸图像。
背景技术
随着近年来通信技术的不断发展,数字化图像作为一种重要的信息传递方式,已经渗透到人们生成生活的方方面面。在视频监控领域中,由于图像采集过程中硬件设备存在噪声、设备分辨率有限和图像传播过程中含有误差等因素的影响,所获得的人脸图像往往分辨率不高,细节信息较差。通过超分辨重建技术可以恢复低分辨人脸图像的纹理细节信息,提升人脸图像的辨识度。现阶段,深度学习由于其强大的非线性能力和特征学习能力,成为了超分辨重建中的主流方法。
中国科学技术大学在其申请的专利文献“一种基于有监督逐像素生成对抗网络的人脸超分辨率方法”(专利申请号:201910879159.8,申请公开号:110689482A)中提出了一种人脸图像超分辨重建方法。该方法设计了一个有监督逐像素的生成对抗网络,包括一个生成器网络和一个逐像素的判别器网络。该方法将高分辨图像下采样得到低分辨图像,将低分辨图像通过生成器生成超分辨图像,接着把超分辨图像和高分辨图像分别输入到逐像素的判别器网络中,得到逐像素判别矩阵,以此计算对抗损失,并将该损失进行反向传播训练参数。该方法采用逐像素判别损失,提高了人脸超分辨重建的相似度和低分辨人脸识别的准确度。但该方法的不足之处是,直接通过下采样得到低分辨人脸图像,这与真实场景中的低分辨人脸图像降质过程不同,使得该方法在应用到真实低分辨人脸图像时,效果较差。
电子科技大学在其申请的专利文献“基于生成对抗网络和亚像素卷积的人脸超分辨率重建方法”(专利申请号:201710347472.8,申请公开号:107154023A)中提出了一种人脸图像超分辨重建方法。该方法同样设计了一个生成对抗网络进行超分辨重建,包括一个生成器网络和一个判别器网络。该方法在生成器网络中通过卷积提取低分辨图像特征,并加入亚像素卷积层来实现低分辨图像特征到高分辨图像的映射,结合生成图像特征的损失信息进行训练。该方法考虑到在低分辨到高分辨的映射过程中,反卷积可能会造成一些图像伪影,因此用亚像素卷积层替代反卷积层后,重建图像整体效果更加自然。但该方法的不足之处是生成器网络中的卷积操作更擅长于提取局部特征,在提取图像全局特征时比较困难,重建过程中未能有效地利用图像的全局整体特征对局部信息进行指导,导致部分重建图像局部不够合理。
发明内容
本发明的目的在于针对上述现有技术存在的不足,提出了一种基于自注意力生成对抗网络的人脸图像超分辨重建方法,用于解决现有方法多针对于简单下采样得到的低分辨图像重建,而对于真实场景低分辨图像重建效果差、纹理细节信息不足的问题。
为实现上述目的,本发明的思路是,构建两个基于自注意力机制的生成对抗网络,在生成器网络中加入自注意力层,在训练网络的损失函数中加入上下文损失,在图像重建过程中加入更多的图像整体全局特征以增强生成图像的细节纹理信息。通过先利用一个生成网络学习低分辨图像的降质过程,再学习超分辨过程,这样的训练方式使得本发明能适用于真实场景中的任意未知降质过程的低分辨图像。
本发明方法的具体步骤包括如下:
(1)构建两个生成器网络:
(1a)搭建两个基本结构相同的生成器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→第1自注意力层→第2卷积层→第2激活函数层→第3卷积层→第3激活函数层→第2自注意力层→第4卷积层→第4激活函数层→输出层;
第1个生成器网络中的残差块组合由12个残差块串联组成;第2个生成器网络中的残差块组合由11个残差块组成,每个残差块的基本结构为:第1归一化层→反卷积层→第1激活函数层→第1卷积层→第2归一化层→池化层→第2激活函数层→第2卷积层;其中,第1个生成器网络中残差块组合中的第1、3、5、7残差块中不设置反卷积层,第2、4、6、8残差块中不设置反卷积层和池化层,第9、10残差块中不设置池化层,第11、12残差块均不设置反卷积层和池化层;第2个生成器网络中的第6、9残差块中不设置池化层,其余残差块中均不设置反卷积层和池化层;将第2个生成器网络残差块组合中的第1至第6残差块组成第一组残差块,第7至第9残差块组成第二组残差块,第10个和第11个残差块组成第三组残差块;残差块的组间采用串联方式连接,每组残差块的内部采用串--跨连接方式连接;
(1b)所述两个生成器网络中的第1激活函数层采用LeakyReLU函数实现,第2、3激活函数层均采用ReLU函数实现,第4激活函数层采用Tanh函数实现;
所述两个生成器网络中的残差块组合的第1、2归一化层均采用BatchNorm2d函数实现,第1、2激活函数均采用ReLU函数实现;
(1c)设置第1个生成器网络的每层参数:
将第1卷积层至第4卷积层的卷积核的个数依次设置为64,96,96,3,卷积核大小均设置为3×3,卷积步长均设置为1;
将第1激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合中第1至第12个残差块的卷积核个数依次设置为64,64,96,96,128,128,256,256,128,128,96,96,将每个残差块中的两个卷积层的卷积核大小均设置为3×3,卷积核的个数相同,卷积步长均设置为1;
将残差块组合中池化层的池化卷积核设置为2×2,池化步长为2;
将残差块组合中的反卷积层的卷积核大小设置为4×4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1;
(1d)设置第2个生成器网络的每层参数:
将第1卷积层至第4卷积层的卷积核的个数依次设置为256,128,96,3,卷积核大小均设置为3×3,卷积步长均设置为1;
将第1激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合的第1组至第3组残差块中的卷积层的卷积核个数依次设置为256,128,96,所有残差块中的卷积层的每个卷积核大小均设置为3×3,卷积步长均设置为1;
将残差块组合中的反卷积层的卷积核设置为4×4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1;
(2)构建判别器网络:
(2a)搭建两个基本结构相同的判别器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→全局平均池化层→第1自注意力层→第2卷积层→第2激活函数层→第2自注意力层→第3卷积层→第3激活函数层Sigmoid→输出层;
所述残差块组合均由6个残差块串联组成,每个残差块的结构为:第1激活函数层→第1卷积层→第2激活函数层→第2卷积层;
所述第1、2激活函数层均采用LeakyReLU函数实现,第3激活函数层采用Sigmoid函数实现;
所述残差块组合中的第1、2激活函数层均采用LeakyReLU函数实现;
(2b)设置判别器网络的每层参数:
第1个判别器网络和第2个判别器网络的参数相同,均按照如下设置;
将第1卷积层的卷积核大小设置为3×3,卷积核的个数设置为64,卷积步长设置为1;将第2、3卷积层的卷积核的个数依次设置为1024,1,卷积核大小均设置为1x1,卷积步长均设置为1;
将所有激活函数层的LeakyReLU函数的斜率均设置为0.2;
将残差块组合中的所有卷积层的卷积核大小均设置为3×3,卷积层的卷积步长均设置为1;
将残差块组合中的所有激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合中的第1至第6个残差块中每2个卷积层中卷积核个数依次设置为:(64,64)、(64,128)、(128,128)、(128,256)、(256,256)、(256,512);
(3)构建两个自注意力生成对抗网络:
将第1个生成器网络和第1个判别器网络级联组成第1个自注意力生成对抗网络,将第2个生成器网络和第2个判别器网络级联组成第2个自注意力生成对抗网络;
(4)初始化自注意力生成对抗网络:
将两个自注意力生成对抗网络中所有的卷积层、反卷积层和归一化层的权重初始化为满足正态分布的随机值,其中:所述正态分布的均值为0,标准差为0.02;
(5)生成初始人脸图像训练数据集:
选取N张大小为W×H的三通道彩色高分辨人脸图像组成初始高分辨图像训练集;再选取M张大小为的三通道彩色低分辨人脸图像组成初始低分辨图像训练集;其中,N>1000,600<M<1000,W、H分别表示每张所选人脸图像的宽和高,64≤W≤256,64≤H≤256,W和H的单位为像素;
(6)训练两个自注意力生成对抗网络:
(6a)将初始高分辨图像训练集输入到第1个生成器网络中,输出对应的预测低分辨图像,利用生成器损失函数计算第1个生成器网络损失值;
利用梯度下降法将第1个生成器网络的损失值反向传播,计算第1个生成器网络每个卷积层和反卷积层的每个卷积核的所有梯度、归一化层的所有梯度、注意力层的所有梯度;
根据第1个生成器网络每个卷积层和反卷积层的每个卷积核的所有梯度、归一化层的所有梯度、注意力层的所有梯度,使用Adam优化器对生成器网络每个卷积层和反卷积层的每个卷积核的所有权重、归一化层的所有权重、注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9);
(6b)将初始低分辨图像训练集中的图像和预测低分辨图像依次输入到第1个判别器网络中,第1个判别器网络输出对应的真伪概率,利用判别器损失函数计算第1个判别器网络的损失值;
利用梯度下降法将第1个判别器网络的损失值反向传播,计算第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度;
根据第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度,使用Adam优化器对判别器网络每个卷积层的每个卷积核的所有权重和自注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9);
(6c)依次重复步骤(6a)和(6b)500次,完成生成对抗网络的训练,得到训练好的第1个生成器网络权重,保存训练好的生成对抗网络中生成器网络每个卷积层和反卷积层的每个卷积核的所有权重、归一化层的所有权重、注意力模块的所有权重;
(6d)第1个生成对抗网络训练结束后,将初始高分辨图像训练集输入到第1生成器网络中,利用保存的生成器网络权重,输出对应的低分辨图像,组成预测低分辨人脸图像训练集;
(6e)将步骤(6d)得到的预测训练集中的低分辨图像输入到第2个生成器网络中,第2个生成器网络输出对应的超分辨图像之后,再将超分辨图像和步骤(6d)得到的初始训练集中的高分辨图像依次输入到第2个判别器网络中,第2个判别器网络对依次输入的图像判别后分别输出对应的真伪概率,采用和步骤(6a)至(6c)中相同的训练方式对第2个生成器网络和第2个判别器网络进行训练,得到训练好的第2个生成器网络并保存训练好的第2个生成器网络权重;
(7)对低分辨人脸图像进行超分辨率重建:
将待超分辨率的低分辨人脸图像输入到第2个生成器网络中,利用保存好的生成器网络权重输出超分辨率重建的结果。
本发明与现有技术相比较,具有以下优点:
第一,本发明提出的基于自注意力生成对抗网络的人脸图像超分辨重建方法,在构建生成器网络中充分考虑到图像的全局整体特征,在生成器输出预测图像过程中和判别器输出预测真伪概率过程中均加入了自注意力层来提取图像全局特征,克服了现有技术中仅通过卷积操作提取图像局部特征的弊端,使得生成的超分辨结果每个像素和所有其它像素都有联系,整体图像更加合理。同时,本发明在构建网络损失函数时引入了上下文损失,克服了现有技术重建得到的高分辨图像中纹理细节信息不足的问题,使得本发明所提出的技术得到的重建图像中纹理细节信息更加丰富,图像视觉效果更好。
第二,本发明构建了两个由生成器网络和判别器网络组成的自注意力生成对抗网络;对于真实场景中未知复杂降质的低分辨图像,首先采用一个生成对抗网络学习低分辨图像的降质过程,再通过另一个生成对抗网络学习超分辨重建过程,这种训练方式克服了现有技术只能针对简单下采样所得的低分辨图像具有良好重建结果、而对于真实场景图像效果较差的问题。本发明能适用于真实场景中任意未知降质过程的低分辨人脸图像。
附图说明
图1为本发明的流程图;
图2为本发明自注意力生成对抗网络中生成器网络的结构示意图;其中,图2(a)为本发明的第1个生成器网络结构示意图,图2(b)为第2个生成器网络结构示意图,图2(c)为生成器中残差块的结构示意图,图2(d)为生成器中注意力层的结构示意图;
图3为本发明自注意力生成对抗网络中判别器网络结构示意图;
图4为本发明自注意力生成对抗网络中判别器网络的残差块图;
图5为本发明的仿真图,其中,图5(a)为输入的低分辨图像,图5(b)为采用Bicubic方法进行重建的结果图,图5(c)为采用SRCNN方法进行重建的结果图,图5(d)为采用SRGAN方法进行重建的结果图,图5(e)为采用本发明方法进行重建的结果图。
具体实施方式
下面结合附图对本发明做进一步的描述。
参照图1,对本发明的具体步骤做进一步的描述。
步骤1.构建两个生成器网络。
搭建两个基本结构相同的生成器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→第1自注意力层→第2卷积层→第2激活函数层→第3卷积层→第3激活函数层→第2自注意力层→第4卷积层→第4激活函数层→输出层。
第1个生成器网络中的残差块组合由12个残差块串联组成;第2个生成器网络中的残差块组合由11个残差块组成。每个残差块的基本结构为:第1归一化层→反卷积层→第1激活函数层→第1卷积层→第2归一化层→池化层→第2激活函数层→第2卷积层;所述残差块的结构参照附图2(c)。
其中,第1个生成器网络中残差块组合中的第1、3、5、7残差块中不设置反卷积层,第2、4、6、8残差块中不设置反卷积层和池化层,第9、10残差块中不设置池化层,第11、12残差块均不设置反卷积层和池化层。第1个生成器网络的具体结构参照图2(a)。
第2个生成器网络中的第6、9残差块中不设置池化层,其余残差块中均不设置反卷积层和池化层;将第2个生成器网络残差块组合中的第1至第6残差块组成第一组残差块,第7至第9残差块组成第二组残差块,第10个和第11个残差块组成第三组残差块;残差块的组间采用串联方式连接,每组残差块的内部采用串--跨连接方式连接;第2个生成器网络的具体结构参照附图2(b)。
第2个生成器网络中,所述的每组残差块内部采用的串--跨连接方式为:每组残差块内从输入到输出有两路连接,一路连接为输入顺序经过各个残差块到达该组输出,为串联方式,一路连接为输入直接连接到输出,为跨连接方式。
所述生成器网络中的自注意力层由4个卷积层组成,具体结构参照图2(d)。
所述两个生成器网络中的第1激活函数层采用LeakyReLU函数实现,第2、3激活函数层均采用ReLU函数实现,第4激活函数层采用Tanh函数实现;所述两个生成器网络中的残差块中的第1、2归一化层均采用BatchNorm2d函数实现,第1、2激活函数均采用ReLU函数实现。
设置第1个生成器网络的每层参数。
将第1卷积层至第4卷积层的卷积核的个数依次设置为64,96,96,3,卷积核大小均设置为3×3,卷积步长均设置为1。
将第1激活函数层的LeakyReLU函数的斜率设置为0.2。
将残差块组合中,第1至第12个残差块的卷积核个数依次设置为64,64,96,96,128,128,256,256,128,128,96,96,将每个残差块中的两个卷积层的卷积核大小均设置为3×3,卷积核的个数相同,卷积步长均设置为1。
将残差块组合中池化层的池化卷积核设置为2×2,池化步长为2。
将残差块组合中的反卷积层的卷积核大小设置为4×4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1。
设置第2个生成器网络的每层参数。
将第1卷积层至第4卷积层的卷积核的个数依次设置为256,128,96,3,卷积核大小均设置为3×3,卷积步长均设置为1。
将第1激活函数层的LeakyReLU函数的斜率设置为0.2。
将残差块组合的第1组至第3组残差块中的卷积层的卷积核个数依次设置为256,128,96,所有残差块中的卷积层的每个卷积核大小均设置为3×3,卷积步长均设置为1。
将残差块组合中的反卷积层的卷积核设置为4x4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1。
步骤2.构建判别器网络。
搭建两个基本结构相同的判别器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→全局平均池化层→第1自注意力层→第2卷积层→第2激活函数层→第2自注意力层→第3卷积层→第3激活函数层→输出层。所述判别器网络的结构参照图3。
所述残差块组合均由6个残差块串联组成,每个残差块的结构为:第1激活函数层→第1卷积层→第2激活函数层→第2卷积层。所述残差块的结构参照图4。
所述第1、2激活函数层均采用LeakyReLU函数实现,第3激活函数层采用Sigmoid函数实现。所述残差块组合中的第1、2激活函数层均采用LeakyReLU函数实现。
设置判别器网络的每层参数。
第1个判别器网络和第2个判别器网络的参数相同,均按照如下设置。
将第1卷积层的卷积核大小设置为3×3,卷积核的个数设置为64,卷积步长设置为1;将第2、3卷积层的卷积核的个数依次设置为1024,1,卷积核大小均设置为1x1,卷积步长均设置为1。
将所有激活函数层的LeakyReLU函数的斜率均设置为0.2。
将残差块组合中的所有卷积层的卷积核大小均设置为3×3,卷积层的卷积步长均设置为1。
将残差块组合中的所有激活函数层的LeakyReLU函数的斜率设置为0.2。
将残差块组合中的第1至第6个残差块中每2个卷积层中卷积核个数依次设置为:(64,64)、(64,128)、(128,128)、(128,256)、(256,256)、(256,512)。
步骤3.构建两个自注意力生成对抗网络。
将第1个生成器网络和第1个判别器网络级联组成第1个自注意力生成对抗网络,将第2个生成器网络和第2个判别器网络级联组成第2个自注意力生成对抗网络。
步骤4.初始化自注意力生成对抗网络。
使用Pytorch函数库中的torch.nn.init.normal函数将两个自注意力生成对抗网络中所有的卷积层、反卷积层和归一化层的权重初始化为满足正态分布的随机值,其中:所述正态分布的均值为0,标准差为0.02;
步骤5.生成初始人脸图像训练数据集。
选取N张大小为W×H的三通道彩色高分辨人脸图像组成初始高分辨图像训练集;再选取M张大小为的三通道彩色低分辨人脸图像组成初始低分辨图像训练集;其中,N>1000,M>800,W、H分别表示每张所选人脸图像的宽和高,64≤W≤256,64≤H≤256,W和H的单位是像素。
步骤6.训练两个自注意力生成对抗网络。
第1步,将初始高分辨图像训练集输入到第1个生成器网络中,输出对应的预测低分辨图像,利用生成器损失函数计算第1个生成器网络损失值。
所述的生成器网络损失函数如下:
LG=10*Lcon+0.05*Ladv+1*Lcx
其中,LG表示生成器网络损失函数,*表示相乘运算,Lcon表示生成器输出的预测低分辨图像和初始低分辨图像间的最小均方误差损失,Ladv表示生成器网络的对抗损失,Lcx表示生成器输出的预测低分辨图像和初始低分辨图像间的上下文损失,其中上下文损失是指将待计算的两张图像分别经过固定的特征提取网络提取一组特征,通过计算两张图像特征间的相似度来衡量两张图像的差异,从而得到的一种损失,称为上下文损失。
分别由下列公式得到:
Ladv=0.5×(D(G(In))-1)2
其中,D(·)表示注意力生成对抗网络中的判别器网络的输出,G(·)表示注意力生成对抗网络中的生成器网络的输出,i、j分别表示图像中像素的位置坐标,表示输入到判别器网络的初始低分辨图像训练集中的图像中坐标为(i、j)处的像素值,表示经生成器输出的预测低分辨图像中坐标为(i、j)处的像素值,In表示输入到生成器网络中的初始高分辨图像训练集中的图像,∑表示求和操作,log表示以10为底的对数操作,N表示输入到判别器的初始低分辨图像训练集中的图像和经生成器输出的预测低分辨图像经过特征提取网络后所得图像特征的维数,max表示取最大值操作,dmn表示初始低分辨图像训练集中图像的第m个特征和生成器输出预测低分辨图像的第n个特征间的余弦距离,dmk表示初始低分辨图像训练集中图像的第m个特征和生成器输出预测低分辨图像的第k个特征间的余弦距离,||·||2表示2范数操作。
利用梯度下降法将第1个判别器网络的损失值反向传播,计算第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度;
根据第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度,使用Adam优化器对判别器网络每个卷积层的每个卷积核的所有权重和自注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9);
第2步,将初始训练集中的低分辨图像和经生成器网络输出的预测低分辨图像依次输入到第1个判别器网络中,第1个判别器网络输出对应的真伪概率,利用判别器损失函数计算第1个判别器网络的损失值。
所述的判决器网络损失函数如下:
LD=0.5×(D(Ic)-1)2+0.5×(D(G(In)))2
其中,LD表示判别器损失函数,Ic表示输入到判别器网络的初始低分辨图像训练集中的图像。
利用梯度下降法将第1个判别器网络的损失值反向传播,计算第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度。
根据第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度,使用Adam优化器对判别器网络每个卷积层的每个卷积核的所有权重和自注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9)。
第3步,依次重复本步骤第1步和第2步500次,完成生成对抗网络的训练,得到训练好的第1个生成器网络权重,保存训练好的生成对抗网络中生成器网络每个卷积层和反卷积层的每个卷积核的所有权重、归一化层的所有权重、注意力模块的所有权重。
第4步,第1个生成对抗网络训练结束后,把初始训练集中的高分辨图像输入到第1生成器网络中,利用保存的生成器网络权重,输出对应的低分辨图像,组成预测低分辨人脸图像训练集。
第5步,将第4步得到的预测训练集中的低分辨图像输入到第2个生成器网络中,第2个生成器网络输出对应的超分辨图像之后,再将超分辨图像和第4步得到的初始训练集中的高分辨图像依次输入到第2个判别器网络中,第2个判别器网络输出对应的真伪概率,采用与本步骤第1步至第3步中相同的训练方式对第2个生成器网络和第2个判别器网络进行训练,得到训练好的第2个生成器网络并保存训练好的第2个生成器网络权重。
步骤7.对低分辨人脸图像进行超分辨率重建。
将待超分辨率的低分辨人脸图像输入到第2个生成器网络中,利用保存好的生成器网络权重输出超分辨率重建的结果。
下面结合仿真实验对本发明的效果做进一步的说明:
1.仿真实验条件:
本发明的仿真实验的硬件平台:处理器为Intel(R)Core i7-8700CPU、主频为3.20GHz、内存为32GB、显卡为GeForce RTX 2080。
本发明的仿真实验的软件平台:Python3.6,Pytorch1.2.0。
2.仿真内容及其结果分析:
本发明的仿真实验是采用本发明的方法,从CelebA人脸图像数据集中选取了60000张大小为64×64的三通道彩色高分辨人脸图像,组成初始高分辨图像训练集;再从Widerface人脸图像数据集中选取了53254张大小为16×16的三通道彩色低分辨人脸图像,组成初始低分辨图像训练集;此外,在Widerface人脸数据集中重新选取了3000张大小为16×16的三通道彩色低分辨人脸图像作为测试集。
为了验证本发明的仿真实验效果,将测试集中所有低分辨人脸图像输入到训练好的第2个生成器网络中进行超分辨重建,得到测试集中所有低分辨人脸图像的超分辨结果。
本发明的仿真实验还采用了现有技术的Bicubic、SRCNN和SRGAN共三种超分辨方法进行仿真,其中,Bicubic为双三次插值法,同过对低分辨图像进行双三次插值得到超分辨图像,SRCNN通过一个三层卷积神经网络对图像进行超分辨重建,首先将低分辨图像通过插值放大,再输入卷积神经网络得到超分辨输出,SRGAN利用一个生成对抗网络对图像进行超分辨重建。本发明的仿真实验将测试集中所有低分辨人脸图像用现有技术的三种对比方法进行仿真实验,再将其结果与本发明的方法进行对比,以此证明本发明方法的有效性。
本发明仿真实验得到的部分重建结果如附图5所示。其中,图5(a)为输入的低分辨图像,其大小为16×16。图5(b)为采用Bicubic方法进行重建所得到的结果图,其大小为64×64。图5(c)为采用SRCNN方法进行重建所得到的结果图,其大小为64×64。图5(d)为采用SRGAN方法进行重建所得到的结果图,其大小为64×64。图5(e)为采用本发明方法进行重建所得到的结果图,其大小为64×64。由图5可以看出,本发明方法所得到的重建结果的视觉效果更好,纹理细节信息更充分。
为了验证本发明的仿真效果,由下式计算PSNR和SSIM,并将计算结果绘成表1:
其中,SSIM(Ic,Ip)表示对输入到公式的初始低分辨训练集中的人脸图像Ic和生成器输出的预测低分辨人脸图像Ip计算得到的结构相似度SSIM值,式中,μc、μp分别为Ic和Ip图像像素值的均值,δc 2、δp 2分别为Ic和Ip图像像素值的方差,δcp为Ic和Ip图像像素值的协方差,c1、c2分别为非常小的常数,避免出现除0操作。PSNR(Ic,Ip)表示对输入到公式的两张人脸图像Ic和Ip计算得到的峰值信噪比PSNR值,lg表示以10为底的对数运算符号。
表1
Bicubic | SRCNN | SRGAN | Ours | |
PSNR | 18.15 | 19.86 | 18.94 | 20.32 |
SSIM | 0.648 | 0.679 | 0.575 | 0.693 |
表1中从左到右依次为Bicubic方法、SRCNN方法、SRGAN方法和本发明的方法。表1中的第一行数据为每种方法得到的峰值信噪比PSNR(Peak Signal to Noise Ratio)值,第2行数据为每种方法得到的结构相似度SSIM(Structural SIMilarity)值。由表1可以看出,本发明方法所得超分辨重建图像的指标更高,说明重建后的图像内容更加准确。
以上仿真实验表明:本发明通过构建两个自注意力生成对抗网络,对低分辨图像的降质过程先进行学习,有效解决了现有技术只能对简单下采样得到的低分辨图像进行重建的不足,本发明可有效用于真实场景中未知复杂降质过程的低分辨图像中,通过在生成器网络和判别器网络中加入自注意力层,解决了现有技术仅利用图像局部信息,生成图像部分不合理的问题,通过引入图像的上下文损失,让它和图像内容损失、对抗损失一起引导生成对抗网络的训练过程,有效提升了重建图像的纹理细节信息,是一种非常实用的人脸图像超分辨重建方法。
Claims (4)
1.一种基于自注意力生成对抗网络的人脸图像超分辨重建方法,其特征在于,构建两个由生成器和判别器组成的自注意力生成对抗网络,通过生成器和判别器的交替更新对生成对抗网络进行训练,首先学习低分辨图像的降质过程,再对低分辨图像进行超分辨重建;该方法具体步骤包括如下:
(1)构建两个生成器网络:
(1a)搭建两个基本结构相同的生成器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→第1自注意力层→第2卷积层→第2激活函数层→第3卷积层→第3激活函数层→第2自注意力层→第4卷积层→第4激活函数层→输出层;
第1个生成器网络中的残差块组合由12个残差块串联组成;第2个生成器网络中的残差块组合由11个残差块组成,每个残差块的基本结构为:第1归一化层→反卷积层→第1激活函数层→第1卷积层→第2归一化层→池化层→第2激活函数层→第2卷积层;其中,第1个生成器网络中残差块组合中的第1、3、5、7残差块中不设置反卷积层,第2、4、6、8残差块中不设置反卷积层和池化层,第9、10残差块中不设置池化层,第11、12残差块均不设置反卷积层和池化层;第2个生成器网络中的第6、9残差块中不设置池化层,其余残差块中均不设置反卷积层和池化层;将第2个生成器网络残差块组合中的第1至第6残差块组成第一组残差块,第7至第9残差块组成第二组残差块,第10个和第11个残差块组成第三组残差块;残差块的组间采用串联方式连接,每组残差块的内部采用串--跨连接方式连接;
(1b)所述两个生成器网络中的第1激活函数层采用LeakyReLU函数实现,第2、3激活函数层均采用ReLU函数实现,第4激活函数层采用Tanh函数实现;
所述两个生成器网络中的残差块组合的第1、2归一化层均采用BatchNorm2d函数实现,第1、2激活函数均采用ReLU函数实现;
(1c)设置第1个生成器网络的每层参数:
将第1卷积层至第4卷积层的卷积核的个数依次设置为64,96,96,3,卷积核大小均设置为3×3,卷积步长均设置为1;
将第1激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合中第1至第12个残差块的卷积核个数依次设置为64,64,96,96,128,128,256,256,128,128,96,96,将每个残差块中的两个卷积层的卷积核大小均设置为3×3,卷积核的个数相同,卷积步长均设置为1;
将残差块组合中池化层的池化卷积核设置为2×2,池化步长为2;
将残差块组合中的反卷积层的卷积核大小设置为4×4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1;
(1d)设置第2个生成器网络的每层参数:
将第1卷积层至第4卷积层的卷积核的个数依次设置为256,128,96,3,卷积核大小均设置为3×3,卷积步长均设置为1;
将第1激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合的第1组至第3组残差块中的卷积层的卷积核个数依次设置为256,128,96,所有残差块中的卷积层的每个卷积核大小均设置为3×3,卷积步长均设置为1;
将残差块组合中的反卷积层的卷积核设置为4×4,卷积核的个数均设置为128,卷积步长设置为2,在反卷积过程中对输入的填充数设置为1;
(2)构建判别器网络:
(2a)搭建两个基本结构相同的判别器网络,每个网络的结构依次为:输入层→第1卷积层→第1激活函数层→残差块组合→全局平均池化层→第1自注意力层→第2卷积层→第2激活函数层→第2自注意力层→第3卷积层→第3激活函数层→输出层;
所述残差块组合均由6个残差块串联组成,每个残差块的结构为:第1激活函数层→第1卷积层→第2激活函数层→第2卷积层;
所述第1、2激活函数层均采用LeakyReLU函数实现,第3激活函数层采用Sigmoid函数实现;
所述残差块组合中的第1、2激活函数层均采用LeakyReLU函数实现;
(2b)设置判别器网络的每层参数:
第1个判别器网络和第2个判别器网络的参数相同,均按照如下设置;
将第1卷积层的卷积核大小设置为3×3,卷积核的个数设置为64,卷积步长设置为1;将第2、3卷积层的卷积核的个数依次设置为1024,1,卷积核大小均设置为1x1,卷积步长均设置为1;
将所有激活函数层的LeakyReLU函数的斜率均设置为0.2;
将残差块组合中的所有卷积层的卷积核大小均设置为3×3,卷积层的卷积步长均设置为1;
将残差块组合中的所有激活函数层的LeakyReLU函数的斜率设置为0.2;
将残差块组合中的第1至第6个残差块中每2个卷积层中卷积核个数依次设置为:(64,64)、(64,128)、(128,128)、(128,256)、(256,256)、(256,512);
(3)构建两个自注意力生成对抗网络:
将第1个生成器网络和第1个判别器网络级联组成第1个自注意力生成对抗网络,将第2个生成器网络和第2个判别器网络级联组成第2个自注意力生成对抗网络;
(4)初始化自注意力生成对抗网络:
将两个自注意力生成对抗网络中所有的卷积层、反卷积层和归一化层的权重初始化为满足正态分布的随机值,其中:所述正态分布的均值为0,标准差为0.02;
(5)生成初始人脸图像训练数据集:
选取N张大小为W×H的三通道彩色高分辨人脸图像组成初始高分辨图像训练集;再选取M张大小为的三通道彩色低分辨人脸图像组成初始低分辨图像训练集;其中,N>1000,M>800,W、H分别表示每张所选人脸图像的宽和高,64≤W≤256,64≤H≤256,W和H的单位为像素;
(6)训练两个自注意力生成对抗网络:
(6a)将初始高分辨图像训练集输入到第1个生成器网络中,输出对应的预测低分辨图像,利用生成器损失函数计算第1个生成器网络损失值;
利用梯度下降法将第1个生成器网络的损失值反向传播,计算第1个生成器网络每个卷积层和反卷积层的每个卷积核的所有梯度、归一化层的所有梯度、注意力层的所有梯度;
根据第1个生成器网络每个卷积层和反卷积层的每个卷积核的所有梯度、归一化层的所有梯度、注意力层的所有梯度,使用Adam优化器对生成器网络每个卷积层和反卷积层的每个卷积核的所有权重、归一化层的所有权重、注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9);
(6b)将初始低分辨图像训练集中的图像和预测低分辨图像依次输入到第1个判别器网络中,第1个判别器网络输出对应的真伪概率,利用判别器损失函数计算第1个判别器网络的损失值;
利用梯度下降法将第1个判别器网络的损失值反向传播,计算第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度;
根据第1个判别器网络每个卷积层的每个卷积核的所有梯度和自注意力层的所有梯度,使用Adam优化器对判别器网络每个卷积层的每个卷积核的所有权重和自注意力层的所有权重进行迭代更新;所述Adam优化器的初始学习率为0.0002,指数衰减率为(0,0.9);
(6c)依次重复步骤(6a)和(6b)500次,完成生成对抗网络的训练,得到训练好的第1个生成器网络权重,保存训练好的生成对抗网络中生成器网络每个卷积层和反卷积层的每个卷积核的所有权重、归一化层的所有权重、注意力模块的所有权重;
(6d)第1个生成对抗网络训练结束后,将初始高分辨图像训练集输入到第1生成器网络中,利用保存的生成器网络权重,输出对应的低分辨图像,组成预测低分辨人脸图像训练集;
(6e)将步骤(6d)得到的预测训练集中的低分辨图像输入到第2个生成器网络中,第2个生成器网络输出对应的超分辨图像之后,再将超分辨图像和步骤(6d)得到的初始训练集中的高分辨图像依次输入到第2个判别器网络中,第2个判别器网络输出对应的真伪概率,采用和步骤(6a)至(6c)中相同的训练方式对第2个生成器网络和第2个判别器网络进行训练,得到训练好的第2个生成器网络并保存训练好的第2个生成器网络权重;
(7)对低分辨人脸图像进行超分辨率重建:
将待超分辨率的低分辨人脸图像输入到第2个生成器网络中,利用保存好的生成器网络权重输出超分辨率重建的结果。
2.根据权利要求1所述的基于自注意力生成对抗网络的人脸图像超分辨重建方法,其特征在于,步骤(1a)中所述每组残差块的内部采用串--跨连接方式连接指的是:第2个生成器网络中每组残差块内部从输入到输出有两路,其中,一路为输入顺序经过各个残差块到达该组输出,为串联方式;一路为输入直接连接到输出,为跨连接方式。
3.根据权利要求1所述的基于自注意力生成对抗网络的人脸图像超分辨重建方法,其特征在于,步骤(6a)中所述的生成器网络损失函数形式如下:
LG=10*Lcon+0.05*Ladv+1*Lcx
其中,LG表示生成器网络损失函数,*表示相乘运算,Lcon表示生成器输出的预测低分辨图像和初始低分辨图像间的最小均方误差损失,Ladv表示生成器网络的对抗损失,Lcx表示生成器输出的预测低分辨图像和初始低分辨图像间的上下文损失,其中上下文损失是指将待计算的两张图像分别经过固定的特征提取网络提取一组特征,通过计算两张图像特征间的相似度来衡量两张图像的差异,从而得到的一种损失,称为上下文损失。分别由下列公式得到:
Ladv=0.5×(D(G(In))-1)2
其中,D(·)表示注意力生成对抗网络中的判别器网络的输出,G(·)表示注意力生成对抗网络中的生成器网络的输出,i、j分别表示图像中像素的位置坐标,表示输入到判别器网络的初始低分辨图像训练集中的图像中坐标为(i、j)处的像素值,表示经生成器输出的预测低分辨图像中坐标为(i、j)处的像素值,In表示输入到生成器网络中的初始高分辨图像训练集中的图像,∑表示求和操作,log表示以10为底的对数操作,N表示输入到判别器的初始低分辨图像训练集中的图像和经生成器输出的预测低分辨图像经过特征提取网络后所得图像特征的维数,max表示取最大值操作,dmn表示初始低分辨图像训练集中图像的第m个特征和生成器输出预测低分辨图像的第n个特征间的余弦距离,dmk表示初始低分辨图像训练集中图像的第m个特征和生成器输出预测低分辨图像的第k个特征间的余弦距离,||·||2表示2范数操作。
4.根据权利要求3中所述的基于自注意力生成对抗网络的人脸图像超分辨重建方法,其特征在于,步骤(6b)中所述的判别器损失函数如下:
LD=0.5×(D(Ic)-1)2+0.5×(D(G(In)))2
其中,LD表示判别器损失函数,Ic表示输入到判别器网络的初始低分辨图像训练集中的图像。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010262877.3A CN111476717B (zh) | 2020-04-07 | 2020-04-07 | 基于自注意力生成对抗网络的人脸图像超分辨重建方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010262877.3A CN111476717B (zh) | 2020-04-07 | 2020-04-07 | 基于自注意力生成对抗网络的人脸图像超分辨重建方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN111476717A true CN111476717A (zh) | 2020-07-31 |
CN111476717B CN111476717B (zh) | 2023-03-24 |
Family
ID=71749833
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010262877.3A Active CN111476717B (zh) | 2020-04-07 | 2020-04-07 | 基于自注意力生成对抗网络的人脸图像超分辨重建方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN111476717B (zh) |
Cited By (24)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112070670A (zh) * | 2020-09-03 | 2020-12-11 | 武汉工程大学 | 全局-局部分离注意力机制的人脸超分辨率方法及系统 |
CN112132741A (zh) * | 2020-08-18 | 2020-12-25 | 山东大学 | 一种人脸照片图像和素描图像的转换方法及系统 |
CN112164122A (zh) * | 2020-10-30 | 2021-01-01 | 哈尔滨理工大学 | 一种基于深度残差生成对抗网络的快速cs-mri重建方法 |
CN112288714A (zh) * | 2020-10-28 | 2021-01-29 | 西安电子科技大学 | 基于深度学习的硬件木马检测方法 |
CN112382338A (zh) * | 2020-11-16 | 2021-02-19 | 南京理工大学 | 基于自注意力残差网络的dna-蛋白质结合位点预测方法 |
CN112396110A (zh) * | 2020-11-20 | 2021-02-23 | 南京大学 | 一种基于生成对抗级联网络增广图像的方法 |
CN112419153A (zh) * | 2020-11-23 | 2021-02-26 | 深圳供电局有限公司 | 图像超分辨率重建方法、装置、计算机设备和存储介质 |
CN112508929A (zh) * | 2020-12-16 | 2021-03-16 | 奥比中光科技集团股份有限公司 | 一种训练生成对抗网络的方法和装置 |
CN112561838A (zh) * | 2020-12-02 | 2021-03-26 | 西安电子科技大学 | 基于残差自注意力和生成对抗网络的图像增强方法 |
CN112580782A (zh) * | 2020-12-14 | 2021-03-30 | 华东理工大学 | 基于通道增强的双注意力生成对抗网络及图像生成方法 |
CN112613435A (zh) * | 2020-12-28 | 2021-04-06 | 杭州魔点科技有限公司 | 人脸图像生成方法、装置、设备及介质 |
CN113096017A (zh) * | 2021-04-14 | 2021-07-09 | 南京林业大学 | 基于深度坐标注意力网络模型的图像超分辨率重建方法 |
CN113240585A (zh) * | 2021-06-01 | 2021-08-10 | 平安科技(深圳)有限公司 | 基于生成对抗网络的图像处理方法、装置及存储介质 |
CN113256494A (zh) * | 2021-06-02 | 2021-08-13 | 同济大学 | 一种文本图像超分辨率方法 |
CN113343705A (zh) * | 2021-04-26 | 2021-09-03 | 山东师范大学 | 一种基于文本语义的细节保持图像生成方法及系统 |
CN113379655A (zh) * | 2021-05-18 | 2021-09-10 | 电子科技大学 | 一种基于动态自注意力生成对抗网络的图像合成方法 |
CN113484908A (zh) * | 2021-08-25 | 2021-10-08 | 成都理工大学 | 部分卷积与注意力机制融合深度学习网络的缺失地震数据重建方法 |
CN113658088A (zh) * | 2021-08-27 | 2021-11-16 | 诺华视创电影科技(江苏)有限公司 | 一种基于多判别器的人脸合成方法及装置 |
CN113744265A (zh) * | 2021-11-02 | 2021-12-03 | 成都东方天呈智能科技有限公司 | 一种基于生成对抗网络的异常检测系统、方法和存储介质 |
CN113780483A (zh) * | 2021-11-12 | 2021-12-10 | 首都医科大学附属北京潞河医院 | 结节超声分类数据处理方法及数据处理系统 |
CN113792862A (zh) * | 2021-09-18 | 2021-12-14 | 中国石油大学(华东) | 基于级联注意力机制的改正单图生成对抗网络的设计方法 |
CN114331840A (zh) * | 2021-12-24 | 2022-04-12 | 汉姆德(宁波)智能医疗科技有限公司 | 重建高保真超分辨显微图像的方法、装置 |
WO2023050587A1 (zh) * | 2021-09-30 | 2023-04-06 | 深圳先进技术研究院 | 一种暗场超分辨率成像方法、模型评估方法及系统 |
CN116030200A (zh) * | 2023-03-27 | 2023-04-28 | 武汉零点视觉数字科技有限公司 | 一种基于视觉融合的场景重构方法与装置 |
Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2019015466A1 (zh) * | 2017-07-17 | 2019-01-24 | 广州广电运通金融电子股份有限公司 | 人证核实的方法及装置 |
CN109859106A (zh) * | 2019-01-28 | 2019-06-07 | 桂林电子科技大学 | 一种基于自注意力的高阶融合网络的图像超分辨率重建方法 |
CN110136063A (zh) * | 2019-05-13 | 2019-08-16 | 南京信息工程大学 | 一种基于条件生成对抗网络的单幅图像超分辨率重建方法 |
CN110706157A (zh) * | 2019-09-18 | 2020-01-17 | 中国科学技术大学 | 一种基于身份先验生成对抗网络的人脸超分辨率重建方法 |
-
2020
- 2020-04-07 CN CN202010262877.3A patent/CN111476717B/zh active Active
Patent Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2019015466A1 (zh) * | 2017-07-17 | 2019-01-24 | 广州广电运通金融电子股份有限公司 | 人证核实的方法及装置 |
CN109859106A (zh) * | 2019-01-28 | 2019-06-07 | 桂林电子科技大学 | 一种基于自注意力的高阶融合网络的图像超分辨率重建方法 |
CN110136063A (zh) * | 2019-05-13 | 2019-08-16 | 南京信息工程大学 | 一种基于条件生成对抗网络的单幅图像超分辨率重建方法 |
CN110706157A (zh) * | 2019-09-18 | 2020-01-17 | 中国科学技术大学 | 一种基于身份先验生成对抗网络的人脸超分辨率重建方法 |
Non-Patent Citations (2)
Title |
---|
丁明航等: "基于注意力生成对抗网络的图像超分辨率重建方法", 《计算机系统应用》 * |
许若波等: "基于组合学习的人脸超分辨率算法", 《计算机应用》 * |
Cited By (39)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112132741A (zh) * | 2020-08-18 | 2020-12-25 | 山东大学 | 一种人脸照片图像和素描图像的转换方法及系统 |
CN112132741B (zh) * | 2020-08-18 | 2023-08-11 | 山东大学 | 一种人脸照片图像和素描图像的转换方法及系统 |
CN112070670B (zh) * | 2020-09-03 | 2022-05-10 | 武汉工程大学 | 全局-局部分离注意力机制的人脸超分辨率方法及系统 |
CN112070670A (zh) * | 2020-09-03 | 2020-12-11 | 武汉工程大学 | 全局-局部分离注意力机制的人脸超分辨率方法及系统 |
CN112288714A (zh) * | 2020-10-28 | 2021-01-29 | 西安电子科技大学 | 基于深度学习的硬件木马检测方法 |
CN112288714B (zh) * | 2020-10-28 | 2022-12-27 | 西安电子科技大学 | 基于深度学习的硬件木马检测方法 |
CN112164122A (zh) * | 2020-10-30 | 2021-01-01 | 哈尔滨理工大学 | 一种基于深度残差生成对抗网络的快速cs-mri重建方法 |
CN112164122B (zh) * | 2020-10-30 | 2022-08-23 | 哈尔滨理工大学 | 一种基于深度残差生成对抗网络的快速cs-mri重建方法 |
CN112382338A (zh) * | 2020-11-16 | 2021-02-19 | 南京理工大学 | 基于自注意力残差网络的dna-蛋白质结合位点预测方法 |
CN112382338B (zh) * | 2020-11-16 | 2022-09-06 | 南京理工大学 | 基于自注意力残差网络的dna-蛋白质结合位点预测方法 |
CN112396110A (zh) * | 2020-11-20 | 2021-02-23 | 南京大学 | 一种基于生成对抗级联网络增广图像的方法 |
CN112396110B (zh) * | 2020-11-20 | 2024-02-02 | 南京大学 | 一种基于生成对抗级联网络增广图像的方法 |
CN112419153A (zh) * | 2020-11-23 | 2021-02-26 | 深圳供电局有限公司 | 图像超分辨率重建方法、装置、计算机设备和存储介质 |
CN112561838B (zh) * | 2020-12-02 | 2024-01-30 | 西安电子科技大学 | 基于残差自注意力和生成对抗网络的图像增强方法 |
CN112561838A (zh) * | 2020-12-02 | 2021-03-26 | 西安电子科技大学 | 基于残差自注意力和生成对抗网络的图像增强方法 |
CN112580782B (zh) * | 2020-12-14 | 2024-02-09 | 华东理工大学 | 基于通道增强的双注意力生成对抗网络及图像生成方法 |
CN112580782A (zh) * | 2020-12-14 | 2021-03-30 | 华东理工大学 | 基于通道增强的双注意力生成对抗网络及图像生成方法 |
CN112508929A (zh) * | 2020-12-16 | 2021-03-16 | 奥比中光科技集团股份有限公司 | 一种训练生成对抗网络的方法和装置 |
CN112613435A (zh) * | 2020-12-28 | 2021-04-06 | 杭州魔点科技有限公司 | 人脸图像生成方法、装置、设备及介质 |
CN113096017A (zh) * | 2021-04-14 | 2021-07-09 | 南京林业大学 | 基于深度坐标注意力网络模型的图像超分辨率重建方法 |
CN113343705A (zh) * | 2021-04-26 | 2021-09-03 | 山东师范大学 | 一种基于文本语义的细节保持图像生成方法及系统 |
CN113379655A (zh) * | 2021-05-18 | 2021-09-10 | 电子科技大学 | 一种基于动态自注意力生成对抗网络的图像合成方法 |
CN113379655B (zh) * | 2021-05-18 | 2022-07-29 | 电子科技大学 | 一种基于动态自注意力生成对抗网络的图像合成方法 |
CN113240585B (zh) * | 2021-06-01 | 2024-05-24 | 平安科技(深圳)有限公司 | 基于生成对抗网络的图像处理方法、装置及存储介质 |
CN113240585A (zh) * | 2021-06-01 | 2021-08-10 | 平安科技(深圳)有限公司 | 基于生成对抗网络的图像处理方法、装置及存储介质 |
CN113256494B (zh) * | 2021-06-02 | 2022-11-11 | 同济大学 | 一种文本图像超分辨率方法 |
CN113256494A (zh) * | 2021-06-02 | 2021-08-13 | 同济大学 | 一种文本图像超分辨率方法 |
CN113484908B (zh) * | 2021-08-25 | 2023-07-14 | 成都理工大学 | 部分卷积与注意力机制融合深度学习网络的缺失地震数据重建方法 |
CN113484908A (zh) * | 2021-08-25 | 2021-10-08 | 成都理工大学 | 部分卷积与注意力机制融合深度学习网络的缺失地震数据重建方法 |
CN113658088A (zh) * | 2021-08-27 | 2021-11-16 | 诺华视创电影科技(江苏)有限公司 | 一种基于多判别器的人脸合成方法及装置 |
CN113792862A (zh) * | 2021-09-18 | 2021-12-14 | 中国石油大学(华东) | 基于级联注意力机制的改正单图生成对抗网络的设计方法 |
CN113792862B (zh) * | 2021-09-18 | 2023-08-22 | 中国石油大学(华东) | 基于级联注意力机制的改正单图生成对抗网络的设计方法 |
WO2023050587A1 (zh) * | 2021-09-30 | 2023-04-06 | 深圳先进技术研究院 | 一种暗场超分辨率成像方法、模型评估方法及系统 |
CN113744265A (zh) * | 2021-11-02 | 2021-12-03 | 成都东方天呈智能科技有限公司 | 一种基于生成对抗网络的异常检测系统、方法和存储介质 |
CN113780483A (zh) * | 2021-11-12 | 2021-12-10 | 首都医科大学附属北京潞河医院 | 结节超声分类数据处理方法及数据处理系统 |
CN113780483B (zh) * | 2021-11-12 | 2022-01-28 | 首都医科大学附属北京潞河医院 | 结节超声分类数据处理方法及数据处理系统 |
CN114331840B (zh) * | 2021-12-24 | 2023-04-07 | 汉姆德(宁波)智能医疗科技有限公司 | 重建高保真超分辨显微图像的方法、装置 |
CN114331840A (zh) * | 2021-12-24 | 2022-04-12 | 汉姆德(宁波)智能医疗科技有限公司 | 重建高保真超分辨显微图像的方法、装置 |
CN116030200A (zh) * | 2023-03-27 | 2023-04-28 | 武汉零点视觉数字科技有限公司 | 一种基于视觉融合的场景重构方法与装置 |
Also Published As
Publication number | Publication date |
---|---|
CN111476717B (zh) | 2023-03-24 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111476717B (zh) | 基于自注意力生成对抗网络的人脸图像超分辨重建方法 | |
CN111768342B (zh) | 基于注意力机制和多级反馈监督的人脸超分辨方法 | |
CN110992275B (zh) | 一种基于生成对抗网络的细化单幅图像去雨方法 | |
CN109035142B (zh) | 一种对抗网络结合航拍图像先验的卫星图像超分辨方法 | |
CN112507997B (zh) | 一种基于多尺度卷积和感受野特征融合的人脸超分辨系统 | |
CN110163813B (zh) | 一种图像去雨方法、装置、可读存储介质及终端设备 | |
CN112001847A (zh) | 相对生成对抗超分辨率重建模型生成高质量图像的方法 | |
CN112767251B (zh) | 基于多尺度细节特征融合神经网络的图像超分辨率方法 | |
CN111429349B (zh) | 基于光谱约束对抗网络的高光谱图像超分辨率方法 | |
CN111915490A (zh) | 一种基于多尺度特征的车牌图像超分辨率重建模型及方法 | |
Wen et al. | Image recovery via transform learning and low-rank modeling: The power of complementary regularizers | |
CN109523470B (zh) | 一种深度图像超分辨率重建方法及系统 | |
CN113284051B (zh) | 一种基于频率分解多注意力机制的人脸超分辨方法 | |
CN110136060B (zh) | 基于浅层密集连接网络的图像超分辨率重建方法 | |
CN111369442B (zh) | 基于模糊核分类与注意力机制的遥感图像超分辨重建方法 | |
Liu et al. | Variational autoencoder for reference based image super-resolution | |
CN109003265B (zh) | 一种基于贝叶斯压缩感知的无参考图像质量客观评价方法 | |
CN110516728B (zh) | 基于去噪卷积神经网络的极化sar地物分类方法 | |
CN112052763A (zh) | 基于双向回顾生成对抗网络的视频异常事件检测方法 | |
CN112580782A (zh) | 基于通道增强的双注意力生成对抗网络及图像生成方法 | |
CN113538246B (zh) | 基于无监督多阶段融合网络的遥感图像超分辨率重建方法 | |
CN110533591B (zh) | 基于编解码器结构的超分辨图像重建方法 | |
CN115880158B (zh) | 一种基于变分自编码的盲图像超分辨率重建方法及系统 | |
CN112950480A (zh) | 一种融合多感受野和密集残差注意的超分辨率重建方法 | |
CN116485741A (zh) | 一种无参考图像质量评价方法、系统、电子设备及存储介质 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |