CN113807395A - 模型训练方法、超分辨率感知方法及相关装置 - Google Patents
模型训练方法、超分辨率感知方法及相关装置 Download PDFInfo
- Publication number
- CN113807395A CN113807395A CN202110915752.0A CN202110915752A CN113807395A CN 113807395 A CN113807395 A CN 113807395A CN 202110915752 A CN202110915752 A CN 202110915752A CN 113807395 A CN113807395 A CN 113807395A
- Authority
- CN
- China
- Prior art keywords
- training
- image
- resolution
- training image
- depth
- 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.)
- Pending
Links
- 238000012549 training Methods 0.000 title claims abstract description 402
- 238000000034 method Methods 0.000 title claims abstract description 76
- 238000013528 artificial neural network Methods 0.000 claims abstract description 45
- 230000008447 perception Effects 0.000 claims abstract description 38
- ORILYTVJVMAKLC-UHFFFAOYSA-N Adamantane Natural products C1C(C2)CC3CC1CC2C3 ORILYTVJVMAKLC-UHFFFAOYSA-N 0.000 claims abstract description 26
- 238000013434 data augmentation Methods 0.000 claims abstract description 20
- 230000003190 augmentative effect Effects 0.000 claims abstract description 14
- 238000000605 extraction Methods 0.000 claims description 47
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 claims description 33
- 230000004913 activation Effects 0.000 claims description 19
- 230000006870 function Effects 0.000 claims description 15
- 238000005070 sampling Methods 0.000 claims description 15
- 238000004590 computer program Methods 0.000 claims description 11
- 238000012545 processing Methods 0.000 claims description 11
- 230000003321 amplification Effects 0.000 claims description 8
- 238000003199 nucleic acid amplification method Methods 0.000 claims description 8
- 238000011176 pooling Methods 0.000 claims description 8
- 238000010586 diagram Methods 0.000 description 10
- 230000008569 process Effects 0.000 description 10
- 230000000694 effects Effects 0.000 description 7
- 238000004422 calculation algorithm Methods 0.000 description 4
- 230000015556 catabolic process Effects 0.000 description 3
- 238000006731 degradation reaction Methods 0.000 description 3
- 230000003287 optical effect Effects 0.000 description 3
- 230000009286 beneficial effect Effects 0.000 description 2
- 238000004364 calculation method Methods 0.000 description 2
- 239000013307 optical fiber Substances 0.000 description 2
- 230000000644 propagated effect Effects 0.000 description 2
- 238000003491 array Methods 0.000 description 1
- 238000004891 communication Methods 0.000 description 1
- 238000012937 correction Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 239000011159 matrix material Substances 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000012544 monitoring process Methods 0.000 description 1
- 230000002093 peripheral effect Effects 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- 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
-
- 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
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- General Health & Medical Sciences (AREA)
- Software Systems (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- Computational Linguistics (AREA)
- Health & Medical Sciences (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Evolutionary Biology (AREA)
- Image Analysis (AREA)
Abstract
本申请提供了一种模型训练方法、超分辨率感知方法及相关装置,所述模型训练方法包括:获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。本申请训练得到的超分辨率感知模型更为稳定且准确率较高。
Description
技术领域
本申请涉及深度学习技术领域,尤其涉及模型训练方法、超分辨率感知方法及相关装置。
背景技术
超分辨率(Super-Resolution)即通过硬件或软件的方法提高原有图像的分辨率,超分辨率重建即通过一系列低分辨率的图像来得到一幅高分辨率的图像。
现有技术CN110111251A公开了一种结合深度监督自编码和感知迭代反投影的图像超分辨率重建方法,包括以下步骤:训练步骤,接收复杂退化条件下的训练图像对训练深度自编码器,将训练完成后的深度自编码器中的编码器的深度卷积神经网络作为学习复杂图像退化模型;重建步骤,将深度自编码器中的编码部分作为迭代反投影算法中的退化模型,使用双三次插值图像作为超分辨率图像迭代初始值,计算超分辨率图像退化后图像与观测图像在特征空间中的感知损失,并用感知损失迭代更新超分辨率图像,直至损失低于阈值后输出当前超分辨率图像作为最终的重建图像。
然而,上述方法在超分辨率重建过程中模型的稳定性较差,准确率较低。
发明内容
本申请的目的在于提供模型训练方法、超分辨率感知方法及相关装置,使训练得到的超分辨率感知模型更为稳定且准确率较高。
本申请的目的采用以下技术方案实现:
第一方面,本申请提供了一种模型训练方法,所述模型训练方法包括:获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。该技术方案的有益效果在于,通过对训练图像进行数据增广,可以使训练数据集尽可能的多样化,使训练得到的超分辨率感知模型具有较强的泛化能力,使用adam优化器训练预设深度神经网络,可以使训练过程更高效,并且梯度相对平滑、稳定,并且,预设深度神经网络采用的是RCAN网络,可以构建非常深的可训练网络并获得较好的图像SR(ImageSuper Resolution,图像超分辨率)性能,综上所述,利用该方法训练得到的超分辨率感知模型更为稳定且准确率较高。
在一些可选的实施例中,所述基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像,包括:获取所述训练图像对应的数据增广类型;基于所述训练图像对应的高分辨率图像和数据增广类型,得到所述新的训练图像对应的高分辨率图像。该技术方案的有益效果在于,根据训练图像对应的数据增广类型,可以对训练图像对应的高分辨率图像执行相应的操作,得到新的训练图像对应的高分辨率图像。
在一些可选的实施例中,所述预设深度神经网络包括浅特征提取模块、深度特征提取模块、上采样模块和重建模块;所述利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,包括:将各所述训练图像输入所述浅特征提取模块,得到各所述训练图像对应的浅特征;将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征;将各所述训练图像对应的深度特征输入所述上采样模块,得到各所述训练图像对应的上采样特征图;将各所述训练图像对应的上采样特征图输入所述重建模块,得到各所述训练图像对应的重建图像;基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像,使用adam优化器对所述预设深度神经网络进行训练,得到超分辨率感知模型。该技术方案的有益效果在于,将训练图像输入浅特征提取模块,得到浅特征,接着将浅特征输入深度特征提取模块,得到深度特征,然后将深度特征输入上采样模块,得到上采样特征图,再将上采样特征图输入重建模块,得到重建图像,最后基于重建图像和高分辨率图像使用adam优化器对预设深度神经网络进行训练,由此得到的超分辨率感知模型的深度较深,可以用于高精度的图像超分辨率。
在一些可选的实施例中,所述浅特征提取模块包括第一卷积层。该技术方案的有益效果在于,可以利用第一卷积层提取训练图像对应的浅特征,参数计算量较小。
在一些可选的实施例中,所述深度特征提取模块包括G个残差组,G是大于1的整数;所述将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征,包括:将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息,i的取值分别是不大于G的每个正整数;将各所述训练图像对应的第G深度中间信息输入第二卷积层,得到各所述训练图像对应的深度卷积中间信息;利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息,获取各所述训练图像对应的深度特征。该技术方案的有益效果在于,利用G个残差组获取第G深度中间信息,再将第G深度中间信息输入第二卷积层,得到深度卷积中间信息,结合浅特征和深度卷积中间信息得到深度特征,一方面,G个残差组可以使训练得到的超分辨率感知模型达到较深的深度,另一方面,深度特征是结合浅特征和深度卷积中间信息二者得到的,浅特征包含丰富的低频信息,深度卷积中间信息是基于第G深度中间信息得到的,包含丰富的残差信息,由于浅特征可以直接应用于获取深度特征,可以使预设深度神经网络专注于学习高频信息,提升超分辨率感知模型的PSNR和SSIM指标。
在一些可选的实施例中,每个残差组均包括B个残差块,B是大于1的整数;所述将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,包括:将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块,得到各所述训练图像对应的第i个残差组的第j残差注意信息,以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息,j的取值分别是不大于B的每个正整数;将各所述训练图像对应的第B残差注意信息输入第三卷积层,得到各所述训练图像对应的注意卷积中间信息;利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息,获取各所述训练图像对应的第i深度中间信息。该技术方案的有益效果在于,利用B个残差块获取第B残差注意信息,将第B残差注意信息输入第三卷积层,得到注意卷积中间信息,结合第i-1深度中间信息和注意卷积中间信息得到第i深度中间信息,由此可以使预设深度神经网络进行残差块之间的残差学习,提升超分辨率感知模型的性能。
在一些可选的实施例中,每个残差块均包括卷积模块和CA模块,各所述残差块均采用如下方法处理输入信息:通过所述卷积模块对所述输入信息顺次进行第一次卷积操作、ReLU激活操作和第二次卷积操作,得到所述输入信息对应的中间信息;通过所述CA模块对所述输入信息对应的中间信息顺次进行全局池化操作、第一次卷积操作、ReLU激活操作、第二次卷积操作和sigmoid激活操作,得到所述输入信息对应的通道权值信息,以及基于所述输入信息对应的通道权值信息和中间信息获取所述输入信息对应的注意力信息;基于所述输入信息和所述输入信息对应的注意力信息,获取所述输入信息对应的残差注意信息。该技术方案的有益效果在于,CA模块可以对不同通道的特征重新进行加权分配,使预设深度神经网络专注于更有用的信道并增强判别学习能力。
在一些可选的实施例中,所述模型训练方法还包括:所述利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,包括:利用所述训练数据集,使用adam优化器和L1损失函数训练所述预设深度神经网络,得到超分辨率感知模型。该技术方案的有益效果在于,模型训练过程中使用的损失函数可以采用L1损失函数,相比于L2损失函数,L1损失函数在达到同样训练效果的情况下所需的迭代次数较少。
第二方面,本申请提供了一种超分辨率感知方法,所述超分辨率感知方法包括:获取待处理图像;将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像;其中,所述超分辨率感知模型是利用上述任一项模型训练方法训练得到的。该技术方案的有益效果在于,将低分辨率较低的待处理图像输入超分辨率感知模型,可以得到高分辨率图像,实现图像超分辨率重建。
第三方面,本申请提供了一种模型训练装置,所述模型训练装置包括:训练数据集部分,用于获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;数据增广部分,用于针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;高分辨率图像部分,用于基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;模型训练部分,用于利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。
在一些可选的实施例中,所述高分辨率图像部分包括:类型获取模块,用于获取所述训练图像对应的数据增广类型;增广图像模块,用于基于所述训练图像对应的高分辨率图像和数据增广类型,得到所述新的训练图像对应的高分辨率图像。
在一些可选的实施例中,所述预设深度神经网络包括浅特征提取模块、深度特征提取模块、上采样模块和重建模块;所述模型训练部分包括:第一提取模块,用于将各所述训练图像输入所述浅特征提取模块,得到各所述训练图像对应的浅特征;第二提取模块,用于将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征;特征图模块,用于将各所述训练图像对应的深度特征输入所述上采样模块,得到各所述训练图像对应的上采样特征图;图像重建模块,用于将各所述训练图像对应的上采样特征图输入所述重建模块,得到各所述训练图像对应的重建图像;训练模块,用于基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像,使用adam优化器对所述预设深度神经网络进行训练,得到超分辨率感知模型。
在一些可选的实施例中,所述浅特征提取模块包括第一卷积层。
在一些可选的实施例中,所述深度特征提取模块包括G个残差组,G是大于1的整数;所述第二提取模块包括:第一信息子模块,用于将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息,i的取值分别是不大于G的每个正整数;第二信息子模块,用于将各所述训练图像对应的第G深度中间信息输入第二卷积层,得到各所述训练图像对应的深度卷积中间信息;深度特征子模块,用于利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息,获取各所述训练图像对应的深度特征。
在一些可选的实施例中,每个残差组均包括B个残差块,B是大于1的整数;所述第一信息子模块包括:注意信息单元,用于将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块,得到各所述训练图像对应的第i个残差组的第j残差注意信息,以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息,j的取值分别是不大于B的每个正整数;卷积信息单元,用于将各所述训练图像对应的第B残差注意信息输入第三卷积层,得到各所述训练图像对应的注意卷积中间信息;中间信息单元,用于利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息,获取各所述训练图像对应的第i深度中间信息。
在一些可选的实施例中,每个残差块均包括卷积模块和CA模块,各所述残差块均采用所述注意信息单元处理输入信息;所述注意信息单元包括:第一处理子单元,用于通过所述卷积模块对所述输入信息顺次进行第一次卷积操作、ReLU激活操作和第二次卷积操作,得到所述输入信息对应的中间信息;第二处理子单元,用于通过所述CA模块对所述输入信息对应的中间信息顺次进行全局池化操作、第一次卷积操作、ReLU激活操作、第二次卷积操作和sigmoid激活操作,得到所述输入信息对应的通道权值信息,以及基于所述输入信息对应的通道权值信息和中间信息获取所述输入信息对应的注意力信息;残差注意子单元,用于基于所述输入信息和所述输入信息对应的注意力信息,获取所述输入信息对应的残差注意信息。
在一些可选的实施例中,所述模型训练部分用于:利用所述训练数据集,使用adam优化器和L1损失函数训练所述预设深度神经网络,得到超分辨率感知模型。
第四方面,本申请提供了一种超分辨率感知装置,所述超分辨率感知装置包括:图像获取模块,用于获取待处理图像;图像处理模块,用于将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像;其中,所述超分辨率感知模型是利用上述任一项模型训练方法训练得到的。
第五方面,本申请提供了一种电子设备,所述电子设备包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现上述任一项模型训练方法的步骤或者上述超分辨率感知方法的步骤。
第六方面,本申请提供了一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现上述任一项模型训练方法的步骤或者上述超分辨率感知方法的步骤。
附图说明
下面结合附图和实施例对本申请进一步说明。
图1是本申请实施例提供的一种模型训练方法的流程示意图;
图2是本申请实施例提供的一种获取高分辨率图像的流程示意图;
图3是本申请实施例提供的一种训练预设深度神经网络的流程示意图;
图4是本申请实施例提供的一种利用深度特征提取模块处理输入信息的流程示意图;
图5是本申请实施例提供的一种得到深度特征的流程示意图;
图6是本申请实施例提供的一种得到第i深度中间信息的流程示意图;
图7是本申请实施例提供的一种利用残差块处理输入信息的流程示意图;
图8是本申请实施例提供的另一种利用残差块处理输入信息的流程示意图;
图9是本申请实施例提供的一种利用CA模块获取注意力信息的流程示意图;
图10是本申请实施例提供的一种超分辨率感知方法的流程示意图;
图11是本申请实施例提供的一种模型训练装置的结构示意图;
图12是本申请实施例提供的一种高分辨率图像部分的结构示意图;
图13是本申请实施例提供的一种模型训练部分的结构示意图;
图14是本申请实施例提供的一种第二提取模块的结构示意图;
图15是本申请实施例提供的一种第一信息子模块的结构示意图;
图16是本申请实施例提供的一种注意信息单元的结构示意图;
图17是本申请实施例提供的一种超分辨率感知装置的结构示意图;
图18是本申请实施例提供的一种电子设备的结构框图;
图19是本申请实施例提供的一种用于实现模型训练方法或者超分辨率感知方法的程序产品的结构示意图。
具体实施方式
下面,结合附图以及具体实施方式,对本申请做进一步描述,需要说明的是,在不相冲突的前提下,以下描述的各实施例之间或各技术特征之间可以任意组合形成新的实施例。
参见图1,本申请提供了一种模型训练方法,所述模型训练方法包括步骤S101~S104。
步骤S101:获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像。具体而言,训练图像对应的高分辨率图像的分辨率大于该训练图像的分辨率。其中,低分辨率的训练图像的分辨率例如是100像素*100像素,高分辨率图像的分辨率例如是800像素*800像素。
步骤S102:针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集。
步骤S103:基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像。
步骤S104:利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN(Residual Channel AttentionNetworks,残差通道注意力网络)网络。
由此,通过对训练图像进行数据增广,可以使训练数据集尽可能的多样化,使训练得到的超分辨率感知模型具有较强的泛化能力,使用adam优化器训练预设深度神经网络,可以使训练过程更高效,并且梯度相对平滑、稳定,并且,预设深度神经网络采用的是RCAN网络,可以构建非常深的可训练网络并获得较好的图像SR(Image Super Resolution,图像超分辨率)性能,综上所述,利用该方法训练得到的超分辨率感知模型更为稳定且准确率较高。
在一具体应用中,adam优化器可以表示为:
mt=β1mt-1+(1-β1)gt
其中,mt和vt分别为t时刻的一阶动量项和二阶动量项,β1、β2为动力值大小(取0.9和0.999),和分别为各自的修正值,Wt表示t时刻即第t迭代模型的参数,gt=△J(Wt)表示t次迭代代价函数关于W的梯度大小,ε是一个取值很小的数(一般为1E-8,为了避免分母为0)。
参见图2,在一些实施方式中,所述步骤S103可以包括步骤S201~S202。
步骤S201:获取所述训练图像对应的数据增广类型。数据增广类型可以包括以下至少一种:水平翻转;垂直翻转;平移;镜像。具体而言,在步骤S201中,训练图像对应的数据增广类型是其中一种。
步骤S202:基于所述训练图像对应的高分辨率图像和数据增广类型,得到所述新的训练图像对应的高分辨率图像。
在一具体应用中,对训练图像进行水平翻转,得到训练图像对应的增广图像并将其作为新的训练图像,对训练图像对应的高分辨率图像也进行水平翻转,得到新的训练图像对应的高分辨率图像。
由此,根据训练图像对应的数据增广类型,可以对训练图像对应的高分辨率图像执行相应的操作,得到新的训练图像对应的高分辨率图像。
参见图3,在一些实施方式中,所述预设深度神经网络可以包括浅特征提取模块、深度特征提取模块(RIR,Residual in Residual)、上采样模块和重建模块,所述步骤S104可以包括步骤S301~S305。
步骤S301:将各所述训练图像输入所述浅特征提取模块,得到各所述训练图像对应的浅特征。
在一些实施方式中,所述浅特征提取模块可以包括第一卷积层。在一具体应用中,第一卷积层的卷积核的大小可以是3×3。
由此,可以利用第一卷积层提取训练图像对应的浅特征,参数计算量较小。
在一具体应用中,利用浅特征提取模块得到各所述训练图像对应的浅特征的表达式如下:
F0=Hsf(ILR)
其中,F0表示训练图像对应的浅特征,Hsf表示卷积运算,ILR表示训练图像。
步骤S302:将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征。
在一具体应用中,利用深度特征提取模块得到各所述训练图像对应的深度特征的表达式如下:
FDF=HRIR(F0)
其中,FDF表示训练图像对应的深度特征,HRIR表示深度特征提取模块。
步骤S303:将各所述训练图像对应的深度特征输入所述上采样模块,得到各所述训练图像对应的上采样特征图。
在一具体应用中,利用上采样模块得到各所述训练图像对应的上采样特征图的表达式如下:
FUP=HUP(FDF)
其中,FUP表示训练图像对应的上采样特征图,HUP表示上采样的算法。
步骤S304:将各所述训练图像对应的上采样特征图输入所述重建模块,得到各所述训练图像对应的重建图像。
在一些实施方式中,所述重建模块可以包括第四卷积层。在一具体应用中,第四卷积层的卷积核的大小可以是3×3。
在一具体应用中,利用重建模块得到各所述训练图像对应的重建图像的表达式如下:
ISR=HREC(FUP)=HRCAN(ILR)
其中,ISR表示训练图像对应的重建图像,HREC表示重建的算法,HRCAN表示超分辨率感知的算法。
步骤S305:基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像,使用adam优化器对所述预设深度神经网络进行训练,得到超分辨率感知模型。
由此,将训练图像输入浅特征提取模块,得到浅特征,接着将浅特征输入深度特征提取模块,得到深度特征,然后将深度特征输入上采样模块,得到上采样特征图,再将上采样特征图输入重建模块,得到重建图像,最后基于重建图像和高分辨率图像使用adam优化器对预设深度神经网络进行训练,由此得到的超分辨率感知模型的深度较深,可以用于高精度的图像超分辨率。
参见图4和图5,在一些实施方式中,所述深度特征提取模块可以包括G个残差组(RG,Residual Group),G是大于1的整数;所述步骤S302可以包括步骤S401~S403。本申请实施例对G的取值不作限定,G例如是10、15或者20。
步骤S401:将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息,i的取值分别是不大于G的每个正整数。
步骤S402:将各所述训练图像对应的第G深度中间信息输入第二卷积层,得到各所述训练图像对应的深度卷积中间信息。第二卷积层的卷积核的大小可以是3×3。
步骤S403:利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息,获取各所述训练图像对应的深度特征。
由此,利用G个残差组获取第G深度中间信息,再将第G深度中间信息输入第二卷积层,得到深度卷积中间信息,结合浅特征和深度卷积中间信息得到深度特征,一方面,G个残差组可以使训练得到的超分辨率感知模型达到较深的深度,另一方面,深度特征是结合浅特征和深度卷积中间信息二者得到的,浅特征包含丰富的低频信息,深度卷积中间信息是基于第G深度中间信息得到的,包含丰富的残差信息,由于浅特征可以直接应用于获取深度特征,可以使预设深度神经网络专注于学习高频信息,提升超分辨率感知模型的PSNR和SSIM指标。
在一具体应用中,第g个残差组的信息流表示如下:
Fg=Hg(Fg-1)=Hg(Hg-1(…H1(F0)…))
其中,g的取值分别是不大于G的每个正整数,Hg表示的是第g个残差组,Fg-1和Fg分别表示第g个残差组的输入信息和输出信息。
在一些实施方式中,所述深度特征提取模块还可以包括长跳跃连接(LSC,Longskip connection),各所述训练图像对应的浅特征可以通过长跳跃连接传输至各所述训练图像对应的深度特征。
利用深度特征提取模块的G个残差组和长跳跃连接得到各所述训练图像对应的深度特征的表达式如下:
FDF=F0+WLSCFG=F0+WLSCHG(HG-1(…H1(F0)…))
其中,WLSC表示第二卷积层的权重,FG表示第G深度中间信息。
参见图4和图6,在一些实施方式中,每个残差组均可以包括B个残差块(RCAB,Residual channel attention block),B是大于1的整数;所述步骤S401可以包括步骤S501~S503。本申请实施例对B的取值不作限定,B例如是20、25或者30。
步骤S501:将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块,得到各所述训练图像对应的第i个残差组的第j残差注意信息,以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息,j的取值分别是不大于B的每个正整数。
步骤S502:将各所述训练图像对应的第B残差注意信息输入第三卷积层,得到各所述训练图像对应的注意卷积中间信息。
步骤S503:利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息,获取各所述训练图像对应的第i深度中间信息。
由此,利用B个残差块获取第B残差注意信息,将第B残差注意信息输入第三卷积层,得到注意卷积中间信息,结合第i-1深度中间信息和注意卷积中间信息得到第i深度中间信息,由此可以使预设深度神经网络进行残差块之间的残差学习,提升超分辨率感知模型的性能。
在一具体应用中,第g个残差组中的第b个残差块的信息流表示如下:
Fg,b=Hg,b(Fg,b-1)=Hg,b(Hg,b-1(…Hg,1(Fg-1)…))
其中,Fg,b-1和Fg,b表示的是第g个残差组中的第b个残差块的输入信息和输出信息,Hg,b表示的是第g个残差组中的第b个残差块。
在一些实施方式中,每个残差组还可以包括短跳跃连接(SSC,Short skipconnection),各所述训练图像对应的第i-1深度中间信息可以通过短跳跃连接传输至各所述训练图像对应的第i深度中间信息。
利用残差组的B个残差块和短跳跃连接得到各所述训练图像对应的第g深度中间信息的表达式如下:
Fg=Fg-1+WgFg,B=Fg-1+WgHg,B(Hg,B-1(…Hg,1(Fg-1)…))
其中,Wg表示第三卷积层的权重。
参见图7和图8,在一些实施方式中,每个残差块均可以包括卷积模块和CA(Channel attention)模块,各所述残差块均可以采用步骤S601~S603处理输入信息。
步骤S601:通过所述卷积模块对所述输入信息顺次进行第一次卷积操作、ReLU激活操作和第二次卷积操作,得到所述输入信息对应的中间信息。
步骤S602:通过所述CA模块对所述输入信息对应的中间信息顺次进行全局池化操作、第一次卷积操作、ReLU激活操作、第二次卷积操作和sigmoid激活操作,得到所述输入信息对应的通道权值信息,以及基于所述输入信息对应的通道权值信息和中间信息获取所述输入信息对应的注意力信息。
参见图9,在一具体应用中,输入信息对应的中间信息对应的特征图的大小为H×W×C。
CA模块对输入信息对应的中间信息进行全局池化操作,得到1×1×C的特征图,对1×1×C的特征图进行第一次卷积操作,得到1×1×C/r的特征图,对1×1×C/r的特征图进行第二次卷积操作,得到1×1×C的特征图,利用sigmoid激活函数进行sigmoid激活操作,得到1×1×C的特征图表示每个通道的权值的描述符,将各通道权重值分别和原特征图对应通道的二维矩阵相乘,得到输入信息对应的注意力信息。
利用CA模块获取所述输入信息对应的注意力信息的表达式如下:
s=f(WUδ(WDz))
其中,z表示全局池化操作,WD表示第一次卷积操作,δ表示ReLU激活操作,WU表示第二次卷积操作,f表示sigmoid激活操作,s表示通道的权重,xc是全局平均池化前输入信息对应的中间信息对应的特征图。
步骤S603:基于所述输入信息和所述输入信息对应的注意力信息,获取所述输入信息对应的残差注意信息。
由此,CA模块可以对不同通道的特征重新进行加权分配,使预设深度神经网络专注于更有用的信道并增强判别学习能力。
在一具体应用中,利用残差块获取输入信息对应的残差注意信息的表达式如下:
Fg,b=Fg,b-1+Rg,b(Xg,b)·Xg,b
Xg,b=W2 g,bδ(W1 g,b·Fg,b-1)
其中,Rg,b表示的是第g个残差组中的第b个残差块的CA模块,Fg,b-1与Fg,b分别表示的是残差块的输入信息和输出信息,Xg,b表示的是输入信息对应的中间信息。
在一些实施方式中,所述步骤S104可以包括:利用所述训练数据集,使用adam优化器和L1损失函数训练所述预设深度神经网络,得到超分辨率感知模型。
由此,模型训练过程中使用的损失函数可以采用L1损失函数,相比于L2损失函数,L1损失函数在达到同样训练效果的情况下所需的迭代次数较少。
参见图10,本申请实施例还提供了一种超分辨率感知方法,所述超分辨率感知方法包括步骤S11~S12。
步骤S11:获取待处理图像。
在一些实施方式中,所述待处理图像可以包括以下至少一种:老照片扫描得到的图像;监控视频中的一帧;存储于医疗机构的细胞组织图像。
步骤S12:将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像。其中,所述超分辨率感知模型是利用上述任一项模型训练方法训练得到的。
由此,将低分辨率较低的待处理图像输入超分辨率感知模型,可以得到高分辨率图像,实现图像超分辨率重建。
在一些实施方式中,通过调整超分辨率感知模型的参数可以使超分辨率感知模型获得不同的解析倍数,例如可以8倍解析待处理图像,16倍解析待处理图像或者48倍解析待处理图像。
参见图11,本申请实施例还提供了一种模型训练装置,其具体实现方式与上述模型训练方法的实施例中记载的实施方式、所达到的技术效果一致,部分内容不再赘述。
所述模型训练装置包括:训练数据集部分101,用于获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;数据增广部分102,用于针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;高分辨率图像部分103,用于基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;模型训练部分104,用于利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。
参见图12,在一些实施方式中,所述高分辨率图像部分103可以包括:类型获取模块201,用于获取所述训练图像对应的数据增广类型;增广图像模块202,用于基于所述训练图像对应的高分辨率图像和数据增广类型,得到所述新的训练图像对应的高分辨率图像。
参见图13,在一些实施方式中,所述预设深度神经网络可以包括浅特征提取模块、深度特征提取模块、上采样模块和重建模块;所述模型训练部分104可以包括:第一提取模块301,用于将各所述训练图像输入所述浅特征提取模块,得到各所述训练图像对应的浅特征;第二提取模块302,用于将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征;特征图模块303,用于将各所述训练图像对应的深度特征输入所述上采样模块,得到各所述训练图像对应的上采样特征图;图像重建模块304,用于将各所述训练图像对应的上采样特征图输入所述重建模块,得到各所述训练图像对应的重建图像;训练模块305,用于基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像,使用adam优化器对所述预设深度神经网络进行训练,得到超分辨率感知模型。
在一些实施方式中,所述浅特征提取模块可以包括第一卷积层。
参见图14,在一些实施方式中,所述深度特征提取模块可以包括G个残差组,G是大于1的整数;所述第二提取模块302可以包括:第一信息子模块401,用于将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息,i的取值分别是不大于G的每个正整数;第二信息子模块402,用于将各所述训练图像对应的第G深度中间信息输入第二卷积层,得到各所述训练图像对应的深度卷积中间信息;深度特征子模块403,用于利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息,获取各所述训练图像对应的深度特征。
参见图15,在一些实施方式中,每个残差组均可以包括B个残差块,B是大于1的整数;所述第一信息子模块401可以包括:注意信息单元501,用于将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块,得到各所述训练图像对应的第i个残差组的第j残差注意信息,以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息,j的取值分别是不大于B的每个正整数;卷积信息单元502,用于将各所述训练图像对应的第B残差注意信息输入第三卷积层,得到各所述训练图像对应的注意卷积中间信息;中间信息单元503,用于利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息,获取各所述训练图像对应的第i深度中间信息。
参见图16,在一些实施方式中,每个残差块均可以包括卷积模块和CA模块,各所述残差块均可以采用所述注意信息单元501处理输入信息;所述注意信息单元501可以包括:第一处理子单元601,用于通过所述卷积模块对所述输入信息顺次进行第一次卷积操作、ReLU激活操作和第二次卷积操作,得到所述输入信息对应的中间信息;第二处理子单元602,用于通过所述CA模块对所述输入信息对应的中间信息顺次进行全局池化操作、第一次卷积操作、ReLU激活操作、第二次卷积操作和sigmoid激活操作,得到所述输入信息对应的通道权值信息,以及基于所述输入信息对应的通道权值信息和中间信息获取所述输入信息对应的注意力信息;残差注意子单元603,用于基于所述输入信息和所述输入信息对应的注意力信息,获取所述输入信息对应的残差注意信息。
在一些实施方式中,所述模型训练部分104可以用于:利用所述训练数据集,使用adam优化器和L1损失函数训练所述预设深度神经网络,得到超分辨率感知模型。
参见图17,本申请实施例还提供了一种超分辨率感知装置,其具体实现方式与上述超分辨率感知方法的实施例中记载的实施方式、所达到的技术效果一致,部分内容不再赘述。
所述超分辨率感知装置包括:图像获取模块11,用于获取待处理图像;图像处理模块12,用于将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像;其中,所述超分辨率感知模型是利用上述任一项模型训练方法训练得到的。
参见图18,本申请实施例还提供了一种电子设备200,电子设备200包括至少一个存储器210、至少一个处理器220以及连接不同平台系统的总线230。
存储器210可以包括易失性存储器形式的可读介质,例如随机存取存储器(RAM)211和/或高速缓存存储器212,还可以进一步包括只读存储器(ROM)213。
其中,存储器210还存储有计算机程序,计算机程序可以被处理器220执行,使得处理器220执行本申请实施例中模型训练方法或者超分辨率感知方法的步骤,其具体实现方式与上述模型训练方法或者超分辨率感知方法的实施例中记载的实施方式、所达到的技术效果一致,部分内容不再赘述。
存储器210还可以包括具有至少一个程序模块215的实用工具214,这样的程序模块215包括但不限于:操作系统、一个或者多个应用程序、其它程序模块以及程序数据,这些示例中的每一个或某种组合中可能包括网络环境的实现。
相应的,处理器220可以执行上述计算机程序,以及可以执行实用工具214。
总线230可以为表示几类总线结构中的一种或多种,包括存储器总线或者存储器控制器、外围总线、图形加速端口、处理器或者使用多种总线结构中的任意总线结构的局域总线。
电子设备200也可以与一个或多个外部设备240例如键盘、指向设备、蓝牙设备等通信,还可与一个或者多个能够与该电子设备200交互的设备通信,和/或与使得该电子设备200能与一个或多个其它计算设备进行通信的任何设备(例如路由器、调制解调器等)通信。这种通信可以通过输入输出接口250进行。并且,电子设备200还可以通过网络适配器260与一个或者多个网络(例如局域网(LAN),广域网(WAN)和/或公共网络,例如因特网)通信。网络适配器260可以通过总线230与电子设备200的其它模块通信。应当明白,尽管图中未示出,可以结合电子设备200使用其它硬件和/或软件模块,包括但不限于:微代码、设备驱动器、冗余处理器、外部磁盘驱动阵列、RAID系统、磁带驱动器以及数据备份存储平台等。
本申请实施例还提供了一种计算机可读存储介质,该计算机可读存储介质用于存储计算机程序,所述计算机程序被执行时实现本申请实施例中模型训练方法或者超分辨率感知方法的步骤,其具体实现方式与上述模型训练方法或者超分辨率感知方法的实施例中记载的实施方式、所达到的技术效果一致,部分内容不再赘述。
图19示出了本实施例提供的用于实现上述模型训练方法或者超分辨率感知方法的程序产品300,其可以采用便携式紧凑盘只读存储器(CD-ROM)并包括程序代码,并可以在终端设备,例如个人电脑上运行。然而,本发明的程序产品300不限于此,在本申请中,可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。程序产品300可以采用一个或多个可读介质的任意组合。可读介质可以是可读信号介质或者可读存储介质。可读存储介质例如可以为但不限于电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。可读存储介质的更具体的例子(非穷举的列表)包括:具有一个或多个导线的电连接、便携式盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑盘只读存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适的组合。
计算机可读存储介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了可读程序代码。这种传播的数据信号可以采用多种形式,包括但不限于电磁信号、光信号或上述的任意合适的组合。可读存储介质还可以是任何可读介质,该可读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。可读存储介质上包含的程序代码可以用任何适当的介质传输,包括但不限于无线、有线、光缆、RF等,或者上述的任意合适的组合。可以以一种或多种程序设计语言的任意组合来编写用于执行本发明操作的程序代码,程序设计语言包括面向对象的程序设计语言诸如Java、C++等,还包括常规的过程式程序设计语言诸如C语言或类似的程序设计语言。程序代码可以完全地在用户计算设备上执行、部分地在用户设备上执行、作为一个独立的软件包执行、部分在用户计算设备上部分在远程计算设备上执行、或者完全在远程计算设备或服务器上执行。在涉及远程计算设备的情形中,远程计算设备可以通过任意种类的网络,包括局域网(LAN)或广域网(WAN),连接到用户计算设备,或者,可以连接到外部计算设备(例如利用因特网服务提供商来通过因特网连接)。
本申请从使用目的上,效能上,进步及新颖性等观点进行阐述,已符合专利法所强调的功能增进及使用要件,本申请以上的说明书及说明书附图,仅为本申请的较佳实施例而已,并非以此局限本申请,因此,凡一切与本申请构造,装置,特征等近似、雷同的,即凡依本申请专利申请范围所作的等同替换或修饰等,皆应属本申请的专利申请保护的范围之内。
Claims (12)
1.一种模型训练方法,其特征在于,所述模型训练方法包括:
获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;
针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;
基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;
利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。
2.根据权利要求1所述的模型训练方法,其特征在于,所述基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像,包括:
获取所述训练图像对应的数据增广类型;
基于所述训练图像对应的高分辨率图像和数据增广类型,得到所述新的训练图像对应的高分辨率图像。
3.根据权利要求1所述的模型训练方法,其特征在于,所述预设深度神经网络包括浅特征提取模块、深度特征提取模块、上采样模块和重建模块;
所述利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,包括:
将各所述训练图像输入所述浅特征提取模块,得到各所述训练图像对应的浅特征;
将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征;
将各所述训练图像对应的深度特征输入所述上采样模块,得到各所述训练图像对应的上采样特征图;
将各所述训练图像对应的上采样特征图输入所述重建模块,得到各所述训练图像对应的重建图像;
基于各所述训练图像对应的重建图像和各所述训练图像对应的高分辨率图像,使用adam优化器对所述预设深度神经网络进行训练,得到超分辨率感知模型。
4.根据权利要求3所述的模型训练方法,其特征在于,所述深度特征提取模块包括G个残差组,G是大于1的整数;
所述将各所述训练图像对应的浅特征输入所述深度特征提取模块,得到各所述训练图像对应的深度特征,包括:
将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,以各所述训练图像对应的浅特征作为各所述训练图像对应的第0深度中间信息,i的取值分别是不大于G的每个正整数;
将各所述训练图像对应的第G深度中间信息输入第二卷积层,得到各所述训练图像对应的深度卷积中间信息;
利用各所述训练图像对应的浅特征和各所述训练图像对应的深度卷积中间信息,获取各所述训练图像对应的深度特征。
5.根据权利要求4所述的模型训练方法,其特征在于,每个残差组均包括B个残差块,B是大于1的整数;
所述将各所述训练图像对应的第i-1深度中间信息输入第i个残差组,得到各所述训练图像对应的第i深度中间信息,包括:
将各所述训练图像对应的第i个残差组的第j-1残差注意信息输入第i个残差组的第j个残差块,得到各所述训练图像对应的第i个残差组的第j残差注意信息,以各所述训练图像对应的第i-1深度中间信息作为各所述训练图像对应的第i个残差组的第0残差注意信息,j的取值分别是不大于B的每个正整数;
将各所述训练图像对应的第B残差注意信息输入第三卷积层,得到各所述训练图像对应的注意卷积中间信息;
利用各所述训练图像对应的第i-1深度中间信息和各所述训练图像对应的注意卷积中间信息,获取各所述训练图像对应的第i深度中间信息。
6.根据权利要求5所述的模型训练方法,其特征在于,每个残差块均包括卷积模块和CA模块,各所述残差块均采用如下方法处理输入信息:
通过所述卷积模块对所述输入信息顺次进行第一次卷积操作、ReLU激活操作和第二次卷积操作,得到所述输入信息对应的中间信息;
通过所述CA模块对所述输入信息对应的中间信息顺次进行全局池化操作、第一次卷积操作、ReLU激活操作、第二次卷积操作和sigmoid激活操作,得到所述输入信息对应的通道权值信息,以及基于所述输入信息对应的通道权值信息和中间信息获取所述输入信息对应的注意力信息;
基于所述输入信息和所述输入信息对应的注意力信息,获取所述输入信息对应的残差注意信息。
7.根据权利要求1所述的模型训练方法,其特征在于,所述利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,包括:
利用所述训练数据集,使用adam优化器和L1损失函数训练所述预设深度神经网络,得到超分辨率感知模型。
8.一种超分辨率感知方法,其特征在于,所述超分辨率感知方法包括:
获取待处理图像;
将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像;
其中,所述超分辨率感知模型是利用权利要求1-7任一项所述的模型训练方法训练得到的。
9.一种模型训练装置,其特征在于,所述模型训练装置包括:
训练数据集部分,用于获取训练数据集,所述训练数据集中的每个训练数据包括一个低分辨率的训练图像和所述训练图像对应的高分辨率图像;
数据增广部分,用于针对至少一个训练图像,对所述训练图像进行数据增广,得到所述训练图像对应的增广图像作为新的训练图像并存储至所述训练数据集;
高分辨率图像部分,用于基于所述训练图像对应的高分辨率图像,获取所述新的训练图像对应的高分辨率图像;
模型训练部分,用于利用所述训练数据集,使用adam优化器训练预设深度神经网络,得到超分辨率感知模型,所述预设深度神经网络采用RCAN网络。
10.一种超分辨率感知装置,其特征在于,所述超分辨率感知装置包括:
图像获取部分,用于获取待处理图像;
图像处理部分,用于将所述待处理图像输入超分辨率感知模型,得到所述待处理图像对应的高分辨率图像;
其中,所述超分辨率感知模型是利用权利要求1-7任一项所述的模型训练方法训练得到的。
11.一种电子设备,其特征在于,所述电子设备包括存储器和处理器,所述存储器存储有计算机程序,所述处理器执行所述计算机程序时实现权利要求1-7任一项所述模型训练方法的步骤或者权利要求8所述超分辨率感知方法的步骤。
12.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现权利要求1-7任一项所述模型训练方法的步骤或者权利要求8所述超分辨率感知方法的步骤。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110915752.0A CN113807395A (zh) | 2021-08-10 | 2021-08-10 | 模型训练方法、超分辨率感知方法及相关装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110915752.0A CN113807395A (zh) | 2021-08-10 | 2021-08-10 | 模型训练方法、超分辨率感知方法及相关装置 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113807395A true CN113807395A (zh) | 2021-12-17 |
Family
ID=78943028
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110915752.0A Pending CN113807395A (zh) | 2021-08-10 | 2021-08-10 | 模型训练方法、超分辨率感知方法及相关装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113807395A (zh) |
Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200334457A1 (en) * | 2019-04-16 | 2020-10-22 | Boe Technology Group Co., Ltd. | Image recognition method and apparatus |
US20200386978A1 (en) * | 2019-06-07 | 2020-12-10 | Drvision Technologies Llc | Artifact regulation methods in deep model training for image transformation |
CN112488924A (zh) * | 2020-12-21 | 2021-03-12 | 深圳大学 | 一种图像超分辨率模型训练方法、重建方法及装置 |
CN112598579A (zh) * | 2020-12-28 | 2021-04-02 | 苏州科达特种视讯有限公司 | 面向监控场景的图像超分辨率方法、装置及存储介质 |
CN112991183A (zh) * | 2021-04-09 | 2021-06-18 | 华南理工大学 | 一种基于多帧注意力机制渐进式融合的视频超分辨率方法 |
CN113096017A (zh) * | 2021-04-14 | 2021-07-09 | 南京林业大学 | 基于深度坐标注意力网络模型的图像超分辨率重建方法 |
-
2021
- 2021-08-10 CN CN202110915752.0A patent/CN113807395A/zh active Pending
Patent Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20200334457A1 (en) * | 2019-04-16 | 2020-10-22 | Boe Technology Group Co., Ltd. | Image recognition method and apparatus |
US20200386978A1 (en) * | 2019-06-07 | 2020-12-10 | Drvision Technologies Llc | Artifact regulation methods in deep model training for image transformation |
CN112488924A (zh) * | 2020-12-21 | 2021-03-12 | 深圳大学 | 一种图像超分辨率模型训练方法、重建方法及装置 |
CN112598579A (zh) * | 2020-12-28 | 2021-04-02 | 苏州科达特种视讯有限公司 | 面向监控场景的图像超分辨率方法、装置及存储介质 |
CN112991183A (zh) * | 2021-04-09 | 2021-06-18 | 华南理工大学 | 一种基于多帧注意力机制渐进式融合的视频超分辨率方法 |
CN113096017A (zh) * | 2021-04-14 | 2021-07-09 | 南京林业大学 | 基于深度坐标注意力网络模型的图像超分辨率重建方法 |
Non-Patent Citations (3)
Title |
---|
ZHANG K 等: "Ntire 2020 challenge on perceptual extreme super-resolution: Methods and results", 《PROCEEDINGS OF THE IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS》, pages 492 - 493 * |
ZHANG YULUN 等: "Image super-resolution using very deep residual channel attention networks", 《PROCEEDINGS OF THE EUROPEAN CONFERENCE ON COMPUTER VISION (ECCV)》, pages 286 - 301 * |
武玉伟 等: "《深度学习基础与应用》", vol. 1, 30 November 2020, 北京理工大学出版社, pages: 147 - 148 * |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111932546A (zh) | 图像分割模型训练方法、图像分割方法、装置、设备及介质 | |
US20200410643A1 (en) | Image generation using subscaling and depth up-scaling | |
CN112950471A (zh) | 视频超分处理方法、装置、超分辨率重建模型、介质 | |
CN112508125A (zh) | 一种图像检测模型的高效全整数量化方法 | |
US20220292638A1 (en) | Video resolution enhancement method, storage medium, and electronic device | |
CN114970513A (zh) | 图像生成方法、装置、设备及存储介质 | |
CN110717958A (zh) | 一种图像重建方法、装置、设备及介质 | |
CN117994708B (zh) | 基于时序一致隐空间引导扩散模型的人体视频生成方法 | |
CN110717405B (zh) | 人脸特征点定位方法、装置、介质及电子设备 | |
CN117422614B (zh) | 基于混合特征交互Transformer的单帧图像超分辨率方法及装置 | |
US10445921B1 (en) | Transferring motion between consecutive frames to a digital image | |
CN113902631A (zh) | 图像处理方法、电子设备和存储介质 | |
CN117094365A (zh) | 图文生成模型的训练方法、装置、电子设备及介质 | |
CN113807395A (zh) | 模型训练方法、超分辨率感知方法及相关装置 | |
CN115601235A (zh) | 一种图像超分辨率网络训练方法、装置、设备及存储介质 | |
CN110728732A (zh) | 一种图像重建方法、装置、设备及介质 | |
CN115760614A (zh) | 图像去噪方法、装置、电子设备及存储介质 | |
CN111798385B (zh) | 图像处理方法及装置、计算机可读介质和电子设备 | |
CN115205117A (zh) | 图像重建方法及装置、计算机存储介质、电子设备 | |
CN111340182B (zh) | 一种输入特征逼近的低复杂度cnn训练方法与装置 | |
CN112668659A (zh) | 模型训练方法、平台和电子设备 | |
CN111062886A (zh) | 酒店图片的超分辨方法、系统、电子产品和介质 | |
CN111950015A (zh) | 一种数据开放输出方法、装置和计算设备 | |
KR20190010489A (ko) | 선택 유닛을 이용한 이미지 처리 방법 및 장치 | |
CN114283060B (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 |