CN113643183A - 非匹配遥感图像弱监督学习超分辨率重建方法及系统 - Google Patents
非匹配遥感图像弱监督学习超分辨率重建方法及系统 Download PDFInfo
- Publication number
- CN113643183A CN113643183A CN202111195140.5A CN202111195140A CN113643183A CN 113643183 A CN113643183 A CN 113643183A CN 202111195140 A CN202111195140 A CN 202111195140A CN 113643183 A CN113643183 A CN 113643183A
- Authority
- CN
- China
- Prior art keywords
- image
- network
- resolution
- loss
- trained
- 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 62
- 238000012549 training Methods 0.000 claims abstract description 73
- 230000006870 function Effects 0.000 claims abstract description 34
- 238000012360 testing method Methods 0.000 claims abstract description 17
- 238000005070 sampling Methods 0.000 claims abstract description 10
- 125000004122 cyclic group Chemical group 0.000 claims abstract description 9
- 238000006731 degradation reaction Methods 0.000 claims description 26
- 230000015556 catabolic process Effects 0.000 claims description 22
- 238000004590 computer program Methods 0.000 claims description 16
- 230000003042 antagnostic effect Effects 0.000 claims description 12
- 238000012545 processing Methods 0.000 claims description 12
- 230000001902 propagating effect Effects 0.000 claims description 11
- 238000000605 extraction Methods 0.000 claims description 10
- 238000003860 storage Methods 0.000 claims description 9
- 230000014759 maintenance of location Effects 0.000 claims description 8
- 230000008485 antagonism Effects 0.000 claims description 6
- 230000008447 perception Effects 0.000 claims description 4
- 238000004321 preservation Methods 0.000 claims description 3
- 230000008569 process Effects 0.000 claims description 3
- 238000010586 diagram Methods 0.000 description 14
- 208000037170 Delayed Emergence from Anesthesia Diseases 0.000 description 7
- 238000006243 chemical reaction Methods 0.000 description 7
- 238000013527 convolutional neural network Methods 0.000 description 3
- 230000006872 improvement Effects 0.000 description 3
- 238000011176 pooling Methods 0.000 description 3
- WDLRUFUQRNWCPK-UHFFFAOYSA-N Tetraxetan Chemical compound OC(=O)CN1CCN(CC(O)=O)CCN(CC(O)=O)CCN(CC(O)=O)CC1 WDLRUFUQRNWCPK-UHFFFAOYSA-N 0.000 description 2
- 238000001514 detection method Methods 0.000 description 2
- 238000009826 distribution Methods 0.000 description 2
- 230000009977 dual effect Effects 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 230000009466 transformation Effects 0.000 description 2
- 230000000007 visual effect Effects 0.000 description 2
- 230000008859 change Effects 0.000 description 1
- 239000002131 composite material Substances 0.000 description 1
- 230000006835 compression Effects 0.000 description 1
- 238000007906 compression Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 238000003708 edge detection Methods 0.000 description 1
- 238000004519 manufacturing process Methods 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 230000011218 segmentation Effects 0.000 description 1
- 239000002904 solvent Substances 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
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/50—Image enhancement or restoration using two or more images, e.g. averaging or subtraction
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Image Analysis (AREA)
- Image Processing (AREA)
Abstract
本发明公开了一种非匹配遥感图像弱监督学习超分辨率重建方法及系统,包括:输入高空间分辨率图像和与高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,结合预设的第一损失函数进行反向传播得到训练好的循环生成对抗网络;对高空间分辨率图像利用双三次插值下采样后输入至训练好的循环生成对抗网络中的第一网络,生成与高空间分辨率图像匹配的第一伪图像;根据低空间分辨率图像、第一伪图像和高空间分辨率图像对预设的超分网络进行训练,结合预设的第二损失函数进行反向传播得到训练好的超分网络;获取低空间分辨率测试图像,输入至训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
Description
技术领域
本发明属于遥感图像处理技术领域,特别涉及一种非匹配遥感图像弱监督学习超分辨率重建方法及系统。
背景技术
遥感图像的空间分辨率越大,图像内容越丰富,能为目标分割、检测等任务提供更有用的信息。但能获取高空间分辨率遥感图像的卫星较少,通过改进卫星硬件设备提升图像的空间分辨率,需要较高的成本。图像超分辨率重建技术可以使用图像处理等软件方式提升图像的分辨率,降低了技术成本,因此遥感图像超分辨率重建具有重要的研究意义。
随着深度学习技术的发展,基于卷积神经网络的超分辨率算法取得了较好的效果。但大部分基于卷积神经网络的超分算法都需要成对匹配的高空间分辨率与低空间分辨率数据进行训练。对于遥感卫星而言,很难获取相同场景下不同空间分辨率的成对图像。现有方法通常使用双三次插值等简单的降质模型合成成对数据集,但这类降质方法并不满足包含噪声、模糊、压缩损失等复杂变化情形下真实的降质过程,在处理真实低空间分辨率遥感图像时效果不佳。
生成对抗网络的方法可以通过生成器与判别器的博弈,生成与目标域相同分布的数据,使重建图像具有良好的视觉效果。但由于缺少先验信息对生成结果进行约束,生成对抗网络的方法易产生假纹理、伪影和不真实的物体等问题,不满足遥感图像中目标的特有形态,对后续的识别、分类等任务造成不好的影响。
在自然图像超分问题中,图像域转换被用于模拟图像退化过程,进而获取匹配的训练数据,为图像超分辨率提供监督。不同于自然图像的域转换问题,由于卫星上遥感传感器设备不同以及遥感图像中主要包含的场景存在巨大差异,不同卫星图像的数据分布差异很大,图像域转换造成合成的低空间分辨率数据出现内容、颜色等偏差。因此,遥感图像域转换需要在学习真实分辨率退化过程的同时,保留输入卫星图像的风格和内容。
发明内容
针对以上技术问题,本发明提供一种非匹配遥感图像弱监督学习超分辨率重建方法及系统。
本发明解决其技术问题采用的技术方案是:
非匹配遥感图像弱监督学习超分辨率重建方法,方法包括以下步骤:
步骤S100:输入高空间分辨率图像和与高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
步骤S200:对高空间分辨率图像利用双三次插值下采样后输入至训练好的循环生成对抗网络中的第一网络,生成与高空间分辨率图像匹配的第一伪图像;
步骤S300:根据与高空间分辨率图像不匹配成对的低空间分辨率图像、第一伪图像和高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
步骤S400:获取低空间分辨率测试图像,将低空间分辨率测试图像输入至训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
步骤S170:根据第一图像、第二伪图像、第二图像、第三图像、第四图像、第五图像和第六图像,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络。
优选地,步骤S170包括:
步骤S173:使用VGG19网络分别对第二伪图像、第一图像、第三图像和低空间分辨率图像进行特征提取,得到第一特征、第二特征、第三特征和第四特征,根据第一特征、第二特征、第三特征和第四特征得到感知损失;
步骤S176:根据循环一致性损失、恒等损失、感知损失、第一网络与第一判别器网络之间的对抗损失和第二网络与第二判别器网络之间的对抗损失得到第一损失值,根据第一损失值对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络。
优选地,根据循环一致性损失、恒等损失、感知损失、第一网络与第一判别器网络之间的对抗损失和第二网络与第二判别器网络之间的对抗损失得到第一损失值具体为:
其中,为预设的第一损失函数的权重参数,,为第一网络,为第二网络,为第一判别器网络,为第二判别器网络,为第一图像, 为第二伪图像,为第二图像,为第三图像,为第四图像,为第五图像,为第六图像,为第一特征、为第二特征、为第三特征和为第四特征,为循环一致性损失,为恒等损失,为感知损失,为对抗损失。
步骤S340:使用边缘提取网络提取第七图像的边缘信息,对低空间分辨率图像进行双三次插值后得到第八图像,根据边缘提取网络提取第八图像的边缘信息,将第八图像的边缘信息作为第四正样本,第七图像的边缘信息作为第四负样本,训练边缘判别网络;
步骤S360:根据伪超分辨率图像、高空间分辨率图像、第八图像的边缘信息、第七图像的边缘信息、低空间分辨率图像和退化图像,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络。
优选地,步骤S360包括:
步骤S365:根据伪图像重建损失、边缘保留损失、退化一致性损失和重建图像的对抗损失得到第二损失值,根据第二损失值对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络。
优选地,步骤S365中根据伪图像重建损失、边缘保留损失、退化一致性损失和重建图像的对抗损失得到第二损失值,具体为:
其中,为预设的第二损失函数的权重参数,,为伪超分辨率图像,为高空间分辨率图像,为第八图像的边缘信息,为第七图像的边缘信息,为低空间分辨率图像,为退化图像,为伪图像重建损失,为退化一致性损失,为边缘保留损失,为重建图像的对抗损失,为超分辨率重建网络,为第三判别网络,为边缘提取网络,为退化网络,为退化判别网络,为边缘判别网络。
非匹配遥感图像弱监督学习超分辨率重建系统,系统包括:
循环生成对抗网络训练模块,用于输入高空间分辨率图像和与高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
第一伪图像生成模块,用于对高空间分辨率图像利用双三次插值下采样后输入至训练好的循环生成对抗网络中的第一网络,生成与高空间分辨率图像匹配的第一伪图像;
超分网络训练模块,用于根据与高空间分辨率图像不匹配成对的低空间分辨率图像、第一伪图像和高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
超分辨率图像生成模块,用于获取低空间分辨率测试图像,将低空间分辨率测试图像输入至训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
计算机设备,包括存储器和处理器,存储器存储有计算机程序,处理器执行计算机程序时实现上述方法的步骤。
计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时实现上述方法的步骤。
上述非匹配遥感图像弱监督学习超分辨率重建方法及系统,首先输入高空间分辨率图像与其不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,通过循环生成对抗网络进行图像域转换,引入预设的第一损失函数,使得循环生成对抗网络的训练更加准确,利用双三次插值下采样与训练好的循环生成对抗网络中的第一网络模型生成与高空间分辨率图像匹配的第一伪图像;采用第一伪图像与匹配的高空间分辨率图像联合监督学习超分网络,使用生成对抗网络作为超分网络中的超分辨率重建网络的基础架构模型,并引入预设的第二损失函数为超分辨率重建网络提供约束,使重建图像具有真实的纹理细节,图像内容能更好的被表达,将低分辨率测试图像输入训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。不依赖成对匹配的高分辨率与低分辨率图像,可以利用非匹配的数据进行遥感图像超分辨率重建,具有较强的灵活性和普适性,重建结果具有较丰富的纹理细节信息以及更好的表达能力。
附图说明
图1为本发明一实施例提供的方法的基本流程示意图;
图2是本发明一实施例提出的残差块示意图;
图3是本发明一实施例提出的判别网络示意图;
图5是本发明一实施例提出的RRDB模块示意图;
图7是本发明一实施例方法和其他现有方法的第一组超分辨率重建结果对比示意图,其中,(a)为BICUBIC方法结果,(b)为ZSSR方法结果,(c)为ESRGAN方法结果, (d)为DRN方法结果,(e)为RRDBNet方法结果,(f)本发明方法结果;
图8是本发明一实施例方法和其他现有方法的第二组超分辨率重建结果对比示意图,其中,(a)为BICUBIC方法结果,(b)为ZSSR方法结果,(c)为ESRGAN方法结果,(d)为DRN方法结果,(e)为RRDBNet方法结果,(f)本发明方法结果。
具体实施方式
为了使本技术领域的人员更好地理解本发明的技术方案,下面结合附图对本发明作进一步的详细说明。
在一个实施例中,如图1所示,非匹配遥感图像弱监督学习超分辨率重建方法,方法包括以下步骤:
步骤S100:输入高空间分辨率图像和与高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
步骤S200:对高空间分辨率图像利用双三次插值下采样后输入至训练好的循环生成对抗网络中的第一网络,生成与高空间分辨率图像匹配的第一伪图像;
步骤S300:根据与高空间分辨率图像不匹配成对的低空间分辨率图像、第一伪图像和高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
步骤S400:获取低空间分辨率测试图像,将低空间分辨率测试图像输入至训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
具体地,输入高空间分辨率图像与其不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,通过循环生成对抗网络进行图像域转换,引入预设的第一损失函数使得循环生成对抗网络的训练更加准确,利用双三次插值下采样与训练好的循环生成对抗网络中的第一网络模型生成与高空间分辨率图像匹配的伪低空间分辨率图像即第一伪图像;采用第一伪图像与匹配的高空间分辨率图像联合监督学习超分网络,使用生成对抗网络作为超分网络中的超分辨率重建网络的基础架构模型,并引入预设的第二损失函数为超分辨率重建网络提供约束,使重建图像具有真实的纹理细节,图像内容能更好的被表达,将低分辨率测试图像输入训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。不依赖成对匹配的高分辨率与低分辨率图像,可以利用非匹配的数据进行遥感图像超分辨率重建,具有较强的灵活性和普适性,重建结果具有较丰富的纹理细节信息以及更好的表达能力。
步骤S170:根据第一图像、第二伪图像、第二图像、第三图像、第四图像、第五图像和第六图像,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络。
具体地,第一网络、第二网络由16个残差块组成,残差块结构如图2所示,残差块包括卷积层COV和Relu层,卷积核的尺寸为,通道数为64。第一判别器网络、第二判别器网络结构如图3所示,包括5层卷积层,四层BN层以及四层Relu层,所有卷积层的卷积核尺寸为,第1、2、3、4层的卷积层通道数分别为64、128、256、512,步长为2,最后一层卷积层通道数为1,步长为1。
在一个实施例中,步骤S170包括:
步骤S173:使用VGG19网络分别对第二伪图像、第一图像、第三图像和低空间分辨率图像进行特征提取,得到第一特征、第二特征、第三特征和第四特征,根据第一特征、第二特征、第三特征和第四特征得到感知损失;
步骤S176:根据循环一致性损失、恒等损失、感知损失、第一网络与第一判别器网络之间的对抗损失和第二网络与第二判别器网络之间的对抗损失得到第一损失值,根据第一损失值对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络。
在一个实施例中,根据循环一致性损失、恒等损失、感知损失、第一网络与第一判别器网络之间的对抗损失和第二网络与第二判别器网络之间的对抗损失得到第一损失值具体为:
其中,为预设的第一损失函数的权重参数,,为第一网络,为第二网络,为第一判别器网络,为第二判别器网络,为第一图像, 为第二伪图像,为第二图像,为第三图像,为第四图像,为第五图像,为第六图像,为第一特征、为第二特征、为第三特征和为第四特征,为循环一致性损失,为恒等损失,为感知损失,为对抗损失。
具体地,为利用卷积神经网络VGG19第五个最大池化层前的第四个卷积层之前的网络进行特征提取。每一轮迭代后,通过反向传播算法训练循环生成对抗网络,重复迭代直至迭代次数达到预设的第一迭代次数阈值时判定完成训练。由于遥感图像的目标形态特殊以及不同遥感图像内容差异较大,直接在遥感图像上应用图像域转换易造成合成数据中目标内容的失真。因此提出了基于感知损失的域转换,即加入了,保留图像中的目标形态不被改变,用于避免生成结果与输入之间产生颜色差异,对抗损失用于使生成网络输出与目标域相同的数据分布,循环一致性损失用于使生成器的输出与输入结构上保持一致,本实施中,预设的第一迭代次数为25000次。
步骤S340:使用边缘提取网络提取第七图像的边缘信息,对低空间分辨率图像进行双三次插值后得到第八图像,根据边缘提取网络提取第八图像的边缘信息,将第八图像的边缘信息作为第四正样本,第七图像的边缘信息作为第四负样本,训练边缘判别网络;
步骤S360:根据伪超分辨率图像高空间分辨率图像、第八图像的边缘信息、第七图像的边缘信息、低空间分辨率图像和退化图像,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络。
在一个实施例中,步骤S360包括:
步骤S365:根据伪图像重建损失、边缘保留损失、退化一致性损失和重建图像的对抗损失得到第二损失值,根据第二损失值对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络。
具体地,本实施中,超分辨率重建网络网络如图4所示,输入先经过一个的卷积层,用于提取低分辨率特征,提取的特征经过23个RDDB模块(密集残差块),得到特征,RDDB模块如图5所示,每个密集残差块由3个稠密块组成,特征经过的卷积层并与特征进行相加,得到特征,使用最近邻上采样将特征进行4倍放大,最后经过2个的卷积层得到最终的输出,其中,RDDB模块中的卷积层卷积核尺寸为。边缘提取网络为DexiNed(边缘提取稠密网络),用于提取与的边缘信息,防止重建图像中出现错误的纹理信息,并能够对图像的纹理、细节信息进行有效表达。退化网络的结构如图6所示,卷积层的卷积核尺寸为,池化层为最大池化,步长为2,残差块如图2所示,退化网络中包含20个残差块。退化判别网络、边缘判别网络的网络结构如图3所示。
在一个实施例中,步骤S365中根据伪图像重建损失、边缘保留损失、退化一致性损失和重建图像的对抗损失得到第二损失值,具体为:
其中,为预设的第二损失函数的权重参数,,为伪超分辨率图像,为高空间分辨率图像,为第八图像的边缘信息,为第七图像的边缘信息,为低空间分辨率图像,为退化图像,为伪图像重建损失,为退化一致性损失,为边缘保留损失,为重建图像的对抗损失,为超分辨率重建网络,为第三判别网络,为边缘提取网络,为退化网络,为退化判别网络,为边缘判别网络。
具体地,每一轮迭代后,通过反向传播算法训练超分网络中的,重复迭代直至迭代次数达到预设的第二迭代次数阈值时判定完成训练。通过退化网络引入退化一致性损失对解空间进行约束,防止超分网络重建出不真实的目标;通过边缘检测网络提取图像边缘信息,增加边缘保留损失的约束,可以保留输入图像的边缘、纹理信息,防止超分网络重建结果中出现假纹理。
在训练和测试阶段,本实施中用到的数据集为DOTA数据集(航空图像目标检测的大规模数据集)与RRSSRD数据集(基于参考图像的遥感超分数据集)。从DOTA数据集中挑选出空间分辨率位于0.1m与0.2m之间的图像,作为高空间分辨率图像,共544张。将RRSSRD中的参考图像(空间分辨率为0.6m)作为低空间分辨率图像,共4046张。
本实施中提出的方法与5种方法进行对比,包括双三次插值BICUBIC、ZSSR(Zero-shot Super Resolution,零样本学习超分辨率网络)、ESRGAN(Enhanced Super-Resolution Generative Adversarial Networks,增强型超分辨率生成对抗网络)、DRN(Dual Regression Networks for Single Image Super-Resolution,双回归超分辨率网络)与RRDBNet(Residual in Residual Dense Network,残差密集超分辨率网络)。具体结果如图7、图8所示,图7为某马路的采样图片,图8为某村庄的采样图片,其中(a)为双三次插值BICUBIC方法结果、 (b)为ZSSR方法结果、 (c)为ESRGAN方法结果、 (d)为DRN方法结果 、(e)为RRDBNet方法结果、 (f)本发明方法结果。从重建结果可以看出,本方法的重建的图像更清晰,目标的边缘、纹理信息较为丰富,具有更好的视觉效果。
和现有技术相比,本发明主要具有以下优点:第一,本发明使用不同卫星拍摄的数据进行遥感图像超分辨率重建,不依赖成对匹配的数据集,降低了对特定降质过程的依赖,具有更强的普适性和灵活性;第二,本发明使用感知损失对图像域转换过程进行改进,在学习真实第一网络的同时,保持合成图像的内容与风格不被改变。第三,本发明通过引入了退化一致性损失和边缘保留损失,对超分网络的生成结果解的空间进行了合理约束,使重建图像具有真实的纹理细节,图像内容能更好的被表达。
在一个实施例中,非匹配遥感图像弱监督学习超分辨率重建系统,系统包括:
循环生成对抗网络训练模块,用于输入高空间分辨率图像和与高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对训练后的循环生成对抗网络进行反向传播,更新训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
伪图像生成模块,用于对高空间分辨率图像利用双三次插值下采样后输入至训练好的循环生成对抗网络中的第一网络,生成与高空间分辨率图像匹配的第一伪图像;
超分网络训练模块,用于根据与高空间分辨率图像不匹配成对的低空间分辨率图像、第一伪图像和高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新训练后的超分网络的网络参数并累计第二迭代次数,当第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
超分辨率图像生成模块,用于获取低空间分辨率测试图像,将低空间分辨率测试图像输入至训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
关于非匹配遥感图像弱监督学习超分辨率重建系统的具体限定可以参见上文中对于非匹配遥感图像弱监督学习超分辨率重建方法的限定,在此不再赘述。上述非匹配遥感图像弱监督学习超分辨率重建系统中的各个模块可全部或部分通过软件、硬件及其组合来实现。上述各模块可以硬件形式内嵌于或独立于计算机设备中的处理器中,也可以以软件形式存储于计算机设备中的存储器中,以便于处理器调用执行以上各个模块对应的操作。
在一个实施例中,一种计算机设备,包括存储器和处理器,存储器存储有计算机程序,处理器执行计算机程序时实现非匹配遥感图像弱监督学习超分辨率重建方法的步骤。
在一个实施例中,一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时实现非匹配遥感图像弱监督学习超分辨率重建方法的步骤。
本领域内的技术人员应明白,本申请的实施例可提供为方法、系统、或计算机程序产品。因此,本申请可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且,本申请可采用在一个或多个其中包含有计算机可用程序代码的计算机可读存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。本申请是参照根据本申请实施例的方法、设备(系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器,使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中,使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品,该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上,使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理,从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。
以上对本发明所提供的一种非匹配遥感图像弱监督学习超分辨率重建及系统进行了详细介绍。本文中应用了具体个例对本发明的原理及实施方式进行了阐述,以上实施例的说明只是用于帮助理解本发明的核心思想。应当指出,对于本技术领域的普通技术人员来说,在不脱离本发明原理的前提下,还可以对本发明进行若干改进和修饰,这些改进和修饰也落入本发明权利要求的保护范围内。
Claims (10)
1.非匹配遥感图像弱监督学习超分辨率重建方法,其特征在于,所述方法包括以下步骤:
步骤S100:输入高空间分辨率图像和与所述高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对所述训练后的循环生成对抗网络进行反向传播,更新所述训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当所述第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
步骤S200:对所述高空间分辨率图像利用双三次插值下采样后输入至所述训练好的循环生成对抗网络中的第一网络,生成与所述高空间分辨率图像匹配的第一伪图像;
步骤S300:根据与所述高空间分辨率图像不匹配成对的低空间分辨率图像、所述第一伪图像和所述高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新所述训练后的超分网络的网络参数并累计第二迭代次数,当所述第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
步骤S400:获取低空间分辨率测试图像,将所述低空间分辨率测试图像输入至所述训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
3.根据权利要求2所述的方法,其特征在于,步骤S170包括:
步骤S173:使用VGG19网络分别对所述第二伪图像、所述第一图像、所述第三图像和所述低空间分辨率图像进行特征提取,得到第一特征、第二特征、第三特征和第四特征,根据所述第一特征、所述第二特征、所述第三特征和所述第四特征得到感知损失;
步骤S340:使用边缘提取网络提取所述第七图像的边缘信息,对所述低空间分辨率图像进行双三次插值后得到第八图像,根据所述边缘提取网络提取所述第八图像的边缘信息,将所述第八图像的边缘信息作为第四正样本,所述第七图像的边缘信息作为第四负样本,训练边缘判别网络;
6.根据权利要求5所述的方法,其特征在于,步骤S360包括:
步骤S365:根据所述伪图像重建损失、所述边缘保留损失、所述退化一致性损失和所述重建图像的对抗损失得到第二损失值,根据所述第二损失值对训练后的超分网络进行反向传播,更新所述训练后的超分网络的网络参数并累计第二迭代次数,当所述第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络。
8.非匹配遥感图像弱监督学习超分辨率重建系统,其特征在于,所述系统包括:
循环生成对抗网络训练模块,用于输入高空间分辨率图像和与所述高空间分辨率图像不匹配成对的低空间分辨率图像,对预设的循环生成对抗网络进行训练,得到训练后的循环生成对抗网络,结合预设的第一损失函数对所述训练后的循环生成对抗网络进行反向传播,更新所述训练后的循环生成对抗网络的网络参数并累计第一迭代次数,当所述第一迭代次数达到预设的第一迭代次数阈值时完成训练,得到训练好的循环生成对抗网络;
伪图像生成模块,用于对所述高空间分辨率图像利用双三次插值下采样后输入至所述训练好的循环生成对抗网络中的第一网络,生成与所述高空间分辨率图像匹配的第一伪图像;
超分网络训练模块,用于根据所述与所述高空间分辨率图像不匹配成对的低空间分辨率图像、所述第一伪图像和所述高空间分辨率图像对预设的超分网络进行训练,得到训练后的超分网络,结合预设的第二损失函数对训练后的超分网络进行反向传播,更新所述训练后的超分网络的网络参数并累计第二迭代次数,当所述第二迭代次数达到预设的第二迭代次数阈值时完成训练,得到训练好的超分网络;
超分辨率图像生成模块,用于获取低空间分辨率测试图像,将所述低空间分辨率测试图像输入至所述训练好的超分网络中的超分辨率重建网络,得到重建后的超分辨率图像。
9.计算机设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1至7中任一项所述的方法的步骤。
10.计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至7中任一项所述的方法的步骤。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111195140.5A CN113643183B (zh) | 2021-10-14 | 2021-10-14 | 非匹配遥感图像弱监督学习超分辨率重建方法及系统 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111195140.5A CN113643183B (zh) | 2021-10-14 | 2021-10-14 | 非匹配遥感图像弱监督学习超分辨率重建方法及系统 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN113643183A true CN113643183A (zh) | 2021-11-12 |
CN113643183B CN113643183B (zh) | 2021-12-21 |
Family
ID=78426749
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202111195140.5A Active CN113643183B (zh) | 2021-10-14 | 2021-10-14 | 非匹配遥感图像弱监督学习超分辨率重建方法及系统 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113643183B (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN115170399A (zh) * | 2022-09-08 | 2022-10-11 | 中国人民解放军国防科技大学 | 多目标场景图像分辨率提升方法、装置、设备和介质 |
CN115936985A (zh) * | 2022-12-01 | 2023-04-07 | 华中光电技术研究所(中国船舶集团有限公司第七一七研究所) | 一种基于高阶退化循环生成对抗网络的图像超分辨率重建方法 |
Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109859107A (zh) * | 2019-02-12 | 2019-06-07 | 广东工业大学 | 遥感图像超分辨率方法、装置、设备及可读存储介质 |
CN110119780A (zh) * | 2019-05-10 | 2019-08-13 | 西北工业大学 | 基于生成对抗网络的高光谱图像超分辨重建方法 |
CN111179172A (zh) * | 2019-12-24 | 2020-05-19 | 浙江大学 | 基于无人机航拍数据的遥感卫星超分辨率实现方法、装置、电子设备及存储介质 |
CN111179177A (zh) * | 2019-12-31 | 2020-05-19 | 深圳市联合视觉创新科技有限公司 | 图像重建模型训练方法、图像重建方法、设备及介质 |
US20200364562A1 (en) * | 2019-05-14 | 2020-11-19 | Robert Bosch Gmbh | Training system for training a generator neural network |
CN113284046A (zh) * | 2021-05-26 | 2021-08-20 | 中国电子科技集团公司第五十四研究所 | 基于无高分辨率参考图的遥感图像增强和复原方法及网络 |
-
2021
- 2021-10-14 CN CN202111195140.5A patent/CN113643183B/zh active Active
Patent Citations (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109859107A (zh) * | 2019-02-12 | 2019-06-07 | 广东工业大学 | 遥感图像超分辨率方法、装置、设备及可读存储介质 |
CN110119780A (zh) * | 2019-05-10 | 2019-08-13 | 西北工业大学 | 基于生成对抗网络的高光谱图像超分辨重建方法 |
US20200364562A1 (en) * | 2019-05-14 | 2020-11-19 | Robert Bosch Gmbh | Training system for training a generator neural network |
CN111179172A (zh) * | 2019-12-24 | 2020-05-19 | 浙江大学 | 基于无人机航拍数据的遥感卫星超分辨率实现方法、装置、电子设备及存储介质 |
CN111179177A (zh) * | 2019-12-31 | 2020-05-19 | 深圳市联合视觉创新科技有限公司 | 图像重建模型训练方法、图像重建方法、设备及介质 |
CN113284046A (zh) * | 2021-05-26 | 2021-08-20 | 中国电子科技集团公司第五十四研究所 | 基于无高分辨率参考图的遥感图像增强和复原方法及网络 |
Non-Patent Citations (2)
Title |
---|
JI, HONG, ET AL: "《Vehicle detection in remote sensing images leveraging on simultaneous super-resolution》", 《IEEE GEOSCIENCE AND REMOTE SENSING LETTERS 17.4》 * |
袁飘逸: "《基于深度学习的图像超分辨率重建算法研究》", 《中国优秀博硕士学位论文全文数据库(硕士)信息科技辑》 * |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN115170399A (zh) * | 2022-09-08 | 2022-10-11 | 中国人民解放军国防科技大学 | 多目标场景图像分辨率提升方法、装置、设备和介质 |
CN115936985A (zh) * | 2022-12-01 | 2023-04-07 | 华中光电技术研究所(中国船舶集团有限公司第七一七研究所) | 一种基于高阶退化循环生成对抗网络的图像超分辨率重建方法 |
Also Published As
Publication number | Publication date |
---|---|
CN113643183B (zh) | 2021-12-21 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Liu et al. | Decoupled spatial-temporal transformer for video inpainting | |
Jiao et al. | Local stereo matching with improved matching cost and disparity refinement | |
Vasluianu et al. | NTIRE 2023 image shadow removal challenge report | |
CN113643183B (zh) | 非匹配遥感图像弱监督学习超分辨率重建方法及系统 | |
Kolesnikov et al. | PixelCNN models with auxiliary variables for natural image modeling | |
Fu et al. | Edge-aware deep image deblurring | |
Zhao et al. | ADRN: Attention-based deep residual network for hyperspectral image denoising | |
Kasem et al. | Spatial transformer generative adversarial network for robust image super-resolution | |
Prajapati et al. | Direct unsupervised super-resolution using generative adversarial network (DUS-GAN) for real-world data | |
Rivadeneira et al. | Thermal image super-resolution challenge-pbvs 2021 | |
CN112836602B (zh) | 基于时空特征融合的行为识别方法、装置、设备及介质 | |
Vasluianu et al. | Shadow removal with paired and unpaired learning | |
CN117635418B (zh) | 生成对抗网络的训练方法、双向图像风格转换方法和装置 | |
CN114283058A (zh) | 基于对抗网络和最大互信息优化的图像超分辨率重建方法 | |
Liu et al. | Facial image inpainting using multi-level generative network | |
Wang et al. | Mars image super-resolution based on generative adversarial network | |
Zhao et al. | Saan: Semantic attention adaptation network for face super-resolution | |
CN112785540B (zh) | 一种扩散加权图像的生成系统及方法 | |
Jiang et al. | Haze relevant feature attention network for single image dehazing | |
CN117237219A (zh) | 一种协同双重协调遥感影像云遮挡缺失重建方法和系统 | |
Paola et al. | Correction of Banding Errors in Satellite Images With Generative Adversarial Networks (GAN) | |
Yang et al. | Underwater image restoration for seafloor targets with hybrid attention mechanisms and conditional generative adversarial network | |
Kulikajevas et al. | Auto-refining 3d mesh reconstruction algorithm from limited angle depth data | |
Jeon et al. | A restoration method of single image super resolution using improved residual learning with squeeze and excitation blocks | |
Dargahi et al. | Single image super-resolution by cascading parallel-structure units through a deep-shallow CNN |
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 |