一种跨分辨率的建筑物影像提取方法及装置

文档序号:30229546发布日期:2022-06-01 04:13阅读:121来源:国知局
一种跨分辨率的建筑物影像提取方法及装置

1.本技术涉及测绘技术领域以及遥感影像处理技术领域,尤其涉及一种跨分辨率的建筑物影像提取方法及装置。


背景技术:

2.快速获取大范围的建筑物分布信息是测绘与遥感领域中最重要的实际生产任务之一。伴随着遥感技术的飞速发展,基于相对高分辨率的遥感影像的地形图勾绘逐渐在快速大范围的中小比例尺测绘中显示出其无可比拟的效率优势。
3.然而,在进行建筑物影像提取时,遥感影像的选取对建筑物影像提取的结果会产生重要影响。在实际测绘生产过程中,遥感影像的选取通常从两个方面考虑:影像质量和获取成本。衡量遥感影像质量一般从遥感影像的物理特征和时间特征出发,包括影像的色调色差、清晰程度、细节表现、空间分辨率和时间分辨率。而获取成本一般主要关注获取影像所花费的时间成本、人力成本和资金成本。虽然对同一套提取算法而言,影像质量越高通常可以带来更好的建筑物提取结果,但是更高质量的影像也会不可避免的需要付出更高的获取成本作为代价。


技术实现要素:

4.本技术提供了一种跨分辨率的建筑物影像提取方法及装置,基于相对低分辨率的遥感影像提取到满足预期的高分辨率要求的建筑物影像。
5.第一方面,本技术提供了一种建筑物影像提取方法。该方法包括:获取低分辨率遥感影像,所述低分辨率遥感影像中包含建筑物;利用生成式对抗网络中的生成器对所述低分辨率遥感影像进行超分辨率重建,确定超分辨率遥感影像和所述建筑物的边缘特征;根据所述边缘特征,利用提取模型在所述超分辨率遥感影像中提取所述建筑物的影像。
6.上述方案通过预先训练的生成器对低分辨率遥感影像进行超分,以获得满足一定超分倍率的高分辨率的遥感影像,从而可以使用预先训练的提取模型在高分辨率的遥感影像中提取到高分辨率的建筑物影像,在满足实际的高质量提取需求的情况下,可以避免增加获取遥感影像的成本。并且,在提取建筑物时,考虑建筑物在超分辨率影像中的边缘特征,可以使提取模型更准确的提取到建筑物影像。
7.在一种可能的实施方式中,所述生成器包括第一网络和第二网络,所述确定超分辨率遥感影像和所述建筑物的边缘特征包括:将所述低分辨率遥感影像输入所述第一网络,根据所述第一网络的输出确定所述超分辨率遥感影像;根据所述建筑物的先验信息,利用所述第二网络在所述超分辨率遥感影像的梯度图中确定所述边缘特征。
8.上述方案通过设计的两阶段的生成器实现对低分辨率遥感影像的超分计算,以获得高分辨率的遥感影像,以及从获得的超分辨率遥感影像的梯度图中提取到建筑物的边缘特征。
9.在一种可能的实施方式中,所述生成式对抗网络通过如下步骤获得:根据超分倍
率,对所述低分辨率遥感影像做降采样处理获得第一遥感影像样本,所述第一遥感影像样本的分辨率小于所述低分辨率遥感影像;根据所述第一遥感影像样本和所述低分辨率遥感影像,更新所述生成式对抗网络中生成器和判别器的参数。
10.其中,超分倍率决定了生成器获得的超分辨率遥感影像的分辨率与低分倍率遥感影像的分辨率的倍数关系。
11.上述方案中使用待提取建筑物影像的低分辨率遥感影像获得的训练样本训练对抗网络的训练样本,一方面可以节省网络的训练成本,另一方面可以避免大量外部数据训练的网络对低分辨率遥感影像不适用,导致生成的高分影像不符合实际情况的问题。
12.在一种可能的实施方式中,在所述根据所述边缘特征,利用提取模型在所述超分辨率遥感影像中提取所述建筑物的影像之前,所述方法还包括:根据所述边缘特征增强所述超分辨率遥感影像中的所述建筑物的边缘区域。
13.在一种可能的实施方式中,所述根据所述边缘特征,利用提取模型在所述超分辨率遥感影像中提取所述建筑物的影像包括:将所述边缘特征和所述超分辨率遥感影像输入所述提取模型,根据所述提取模型的输出,确定所述建筑物的影像。
14.在一种可能的实施方式中,所述提取模型通过如下步骤训练获得:获取第二遥感影像样本以及所述第二遥感影像样本的梯度图;根据所述第二遥感影像样本、所述第二遥感影像样本的梯度图、以及所述第二遥感影像样本对应的建筑物标签,更新所述提取模型的参数。
15.在一种可能的实施方式中,所述第二遥感影像样本以及所述第二遥感影像样本的梯度图通过所述生成器获得。
16.第二方面,本技术还提供一种建筑物影像提取装置。该装置包括:获取模块、超分模块和提取模块。
17.其中,获取模块用于获取低分辨率遥感影像,所述低分辨率遥感影像中包含建筑物。
18.其中,超分模块用于利用生成式对抗网络中的生成器对所述低分辨率遥感影像进行超分辨率重建,确定超分辨率遥感影像和所述建筑物的边缘特征。
19.其中,提取模块用于根据所述边缘特征,利用提取模型在所述超分辨率遥感影像中提取所述建筑物的影像。
20.在一种可能的实施方式中,所述生成器包括第一网络和第二网络,所述超分模块具体用于:将所述低分辨率遥感影像输入所述第一网络,根据所述第一网络的输出确定所述超分辨率遥感影像;根据所述建筑物的先验信息,利用所述第二网络在所述超分辨率遥感影像的梯度图中确定所述边缘特征。
21.在一种可能的实施方式中,所述超分模块还用于通过如下步骤获得生成式对抗网络:根据超分倍率,对所述低分辨率遥感影像做降采样处理获得第一遥感影像样本,所述第一遥感影像样本的分辨率小于所述低分辨率遥感影像;根据所述第一遥感影像样本和所述低分辨率遥感影像,更新所述生成式对抗网络中生成器和判别器的参数。
22.在一种可能的实施方式中,所述超分模块还用于:根据所述边缘特征增强所述超分辨率遥感影像中的所述建筑物的边缘区域。
23.在一种可能的实施方式中,所述提取模块具体用于:将所述边缘特征和所述超分
辨率遥感影像输入所述提取模型,根据所述提取模型的输出,确定所述建筑物的影像。
24.在一种可能的实施方式中,所述提取模块还通过如下步骤训练获得提取模型:获取第二遥感影像样本以及所述第二遥感影像样本的梯度图;根据所述第二遥感影像样本、所述第二遥感影像样本的梯度图、以及所述第二遥感影像样本对应的建筑物标签,更新所述提取模型的参数。
25.在一种可能的实施方式中,所述第二遥感影像样本以及所述第二遥感影像样本的梯度图通过所述生成器获得。
26.第三方面,本技术还提供一种计算设备。包括:处理器和存储器,所述处理器用于执行存储于所述存储器内的计算机程序以实现前述第一方面或第一方面的任意一种可能的实现方式所提供的提取方法。
27.第四方面,本技术还提供一种计算机可读存储介质。包括指令,当所述指令在计算设备上运行时,使得所述计算设备执行如前述第一方面或第一方面的任意一种可能的实现方式所提供的提取方法。
28.第五方面,本技术还提供计算机程序产品,当其在计算机上运行时,使得计算机执行上述第一方面或第一方面的任意一种可能的实现方式所提供的提取方法。
29.上述提供的任一种装置或计算机存储介质或计算机程序产品,均用于执行上文所提供的方法,因此,其所能达到的有益效果可参考上文提供的对应方法中的对应方案的有益效果,此处不再赘述。
附图说明
30.图1是本技术实施例提供的一种提取建筑物的整体流程图;
31.图2是本技术实施例提供的一种建筑物提取方法的流程图;
32.图3是本技术实施例提供的一种生成式对抗网络的结构示意图;
33.图4是本技术实施例提供的一种生成式对抗网络的训练方法的流程图;
34.图5是本技术实施例提供的一种获取生成式对抗网络的训练样本的示意图;
35.图6是本技术实施例提供的一种提取模型的训练方法的流程图;
36.图7是本技术实施例提供的一种建筑物提取装置的结构示意图;
37.图8是本技术实施例提供的一种提取设备的结构示意图。
具体实施方式
38.为了使本技术实施例的目的、技术方案和优点更加清楚,下面将结合附图,对本技术实施例中的技术方案进行描述。
39.在本技术实施例的描述中,“示例性的”、“例如”或者“举例来说”等词用于表示作例子、例证或说明。本技术实施例中被描述为“示例性的”、“例如”或者“举例来说”的任何实施例或设计方案不应被解释为比其它实施例或设计方案更优选或更具优势。确切而言,使用“示例性的”、“例如”或者“举例来说”等词旨在以具体方式呈现相关概念。
40.在本技术实施例的描述中,术语“和/或”,仅仅是一种描述关联对象的关联关系,表示可以存在三种关系,例如,a和/或b,可以表示:单独存在a,单独存在b,同时存在a和b这三种情况。另外,除非另有说明,术语“多个”的含义是指两个或两个以上。例如,多个系统是
指两个或两个以上的系统,多个屏幕终端是指两个或两个以上的屏幕终端。
41.此外,术语“第一”、“第二”仅用于描述目的,而不能理解为指示或暗示相对重要性或者隐含指明所指示的技术特征。由此,限定有“第一”、“第二”的特征可以明示或者隐含地包括一个或者更多个该特征。术语“包括”、“包含”、“具有”及它们的变形都意味着“包括但不限于”,除非是以其他方式另外特别强调。
42.在实际的测绘应用中,如背景技术的介绍,从遥感影像中提取建筑物影像时,需要考虑建筑物影像的质量需求与遥感影像的获取成本之前的关系。
43.同一种建筑物,在不同分辨率的遥感影像中所呈现出的颜色、纹理特征、边缘细节乃至特征数量都存在巨大的差异。为了满足建筑物影像的高质量需求,可以使用提取模型从高分辨率遥感影像中提取建筑物影像,但获取原始的高分辨率遥感影像会耗费较大的成本。
44.如果使用提取模型从低分辨率遥感影像中提取建筑物影像,虽然可以避免获取高分辨率遥感影像导致的成本过大问题,但是,该方案获取的建筑物影像中建筑物的边缘轮廓和范围细节较差,无法满足高质量提取需求。
45.鉴于此,本技术实施例提供一种跨分辨率的建筑物影像提取方法。如图1所示的建筑物提取的整体流程图,该方法包括超分阶段和分割阶段。
46.在超分阶段,基于包含建筑物的低分辨率遥感影像(low resolution images,lr images或者i
low
),使用数据集制作模块(ipg)获得超分辨率增强模块(easr)的训练样本并进行训练。其中,easr采用生成式对抗网络的结构。easr训练结束之后,使用训练好的生成式对抗网络对低分辨率遥感影像进行超分辨率重建,获得的超分辨率遥感影像(super resolution images or super resolved images,sr images或者i
sr
)以及建筑物的边缘特征
47.在分割阶段,将边缘特征作为参考,将其与i
sr
输入提取模型中,以提取建筑物的影像。其中,提取模型即为训练后的分割网络(des)。
48.该方法基于低分辨率遥感影像实现高质量的建筑物影像提取,不仅满足了建筑物影像的高质量需求,还可以避免增大遥感影像的获取成本。并且,在提取时,结合建筑物的边缘特征还可以提高提取建筑物影像的准确度。
49.下面结合附图2,对本技术实施例的建筑物影像提取方法做具体介绍。
50.图2是本技术实施例提供的一种跨分辨率的建筑物影像提取方法的流程图。该方法可以应用于提取设备中。该提取设备可以实现建筑物影像提取。
51.如图2所示,该方法包括如下的步骤s201-步骤s203。
52.在步骤s201中,获取低分辨率遥感影像。
53.在进行地形图勾绘时,提取设备可以从数据库中获取已有的包含建筑物的低分辨率遥感影像。其中,低分辨率遥感影像可以是由卫星采集获得的影像。
54.在步骤s202中,利用生成式对抗网络中的生成器对该低分辨率遥感影像进行超分辨率重建,获得超分辨率遥感影像以及建筑物的边缘特征。该超分辨率遥感影像相当于该低分辨率遥感影像指示的区域的高分辨率遥感影像。
55.生成式对抗网络包括生成器和判别器。该生成式对抗网络根据低分辨率遥感影像
训练获得。如图1所示,使用ipg模块处理低分辨率遥感影像,获得生成式对抗网络的训练样本。在生成式对抗网络训练结束之后,其中的生成器即为easr。
56.提取设备可以将低分辨率遥感影像输入训练好的生成器中,根据生成器的输出确定超分辨率遥感影像和建筑物的边缘特征。生成器对输入的影像进行分层特征提取和编码,获得底层到高层的语义特征,从而构建出具备更好的细节表现的高分辨率的遥感影像。在这一过程中,生成器先对输入的数据逐层级进行编码,随后对获得的高级语义信息进行解码,从而重构出分辨率更高的影像。此外,为了增强生成器对边缘细节的关注,还让生成器使用影像的梯度信息指导边缘细节的纹理恢复,而不是保守的抑制掉大部分边缘细节特征,从而增强生成器的鲁棒性。
57.具体地,参阅图3所示的生成式对抗网络的结构示意图,生成器(generator)可以包括第一网络和第二网络。继续参阅图3,提取设备可以使用第一网络对低分辨率遥感影像进行超分辨率重建为超分辨率遥感影像。提取设备根据建筑物的先验信息,使用第二网络从超分辨率遥感影像的梯度图中提取增强的高频信息(enhanced high frequency information),即建筑物的边缘特征。
58.生成式对抗网络的具体结构以及训练过程将在后文中结合附图3做具体介绍,此处不再赘述。
59.在一个示例中,为了提高提取建筑物的准确性,可以使用前述获得的建筑物的边缘特征对超分辨率遥感影像进行增强处理,以增强建筑物在超分辨率遥感影像中建筑物的边缘部分。如图3所示,可以在初始获得的超分辨率遥感影像(sr base images)中去除该超分辨率遥感影像的梯度图(grad of sr base images)之后,添加建筑物的边缘特征,获得增强后的超分辨率遥感影像(sr images)。
60.在步骤s203中,根据边缘特征,利用提取模型在高分辨率遥感影像中提取建筑物的影像。
61.提取设备在获得超分辨率遥感影像之后,如图1所示,可以将该超分辨率遥感影像和建筑物的边缘特征输入提取模型,获得提取模型对输入数据进行特征编码、融合和解码后的输出,根据提取模型的输出确定建筑物的影像。
62.其中,提取模型采用双路分割网络的结构。双路主要指的是影像和边缘特征将分别通过不同的特征提取器进行从低层到高层的特征提取,针对提取到的双路特征,经由特征融合之后建筑物的边缘区域将得到特定的特征增强,其对应的语义信息等得到相应的放大,并抑制掉建筑物周围无关和冗余的信息,从而获得相对高分辨率的建筑物提取结果。提取模型的具体结构介绍以及训练过程将在后文中结合附图进行详细介绍,此处不再赘述。
63.在一个示例中,提取设备在将该超分辨率遥感影像输入提取模型时,还可以使用建筑物的边缘特征对超分辨率遥感影像进行特征增强处理,以增强建筑物在超分辨率遥感影像中的边缘部分。
64.上述方法实施例,一方面使用生成器对低分辨率遥感影像进行超分,获得相对高的高分辨率遥感影像,可以在不增加遥感影像的获取成本的情况下,由低分辨率遥感影像提取到高分辨率的建筑物影像,满足实际高质量的提取需求。另一方面,在提取建筑物影像时,融合高分影像和建筑物的边缘特征,可以提高建筑物提取结果的边界准确性和抑制建筑物周边的冗余信息。
65.基于上述图2所示的建筑物影像的提取方法实施例,本技术实施例提供一种生成式对抗网络的训练方法。
66.该生成式对抗网络包括生成器和判别器。该生成器用于对低分辨率遥感影像进行超分辨率重建,以获得高分辨率遥感影像。参阅图2所示的生成式对抗网络的结构示意图,该生成器具体可以包括第一网络和第二网络。第一网络可以采用残差神经网络,包括卷积层(conv layer)、激活层(leakyrelu layer)、残差模块和解码层(decoder),残差模块(residual block)采用残差链接(skip connection)结构。第二网络可以采用包括卷积层(conv layer)、激活层(leakyrelu layer)、两个分支(frame branch和mask branch)的神经网络结构。该判别器为包括多个卷积层、多个激活层、多个归一化层(batch normalization layer)和全连接层(dense)的神经网络。
67.图4是本技术实施例提供的一种生成式对抗网络的训练方法的流程图。该方法可以应用于上述的提取设备。
68.如图4所示,该方法包括如下的步骤s401和步骤s402。
69.在步骤s401中,根据低分辨率遥感影像确定训练样本。
70.训练样本包括低分辨率遥感影像对应的初始影像切片库、第一遥感影像样本对应的初始影像配对库和低分辨率遥感影像对应的建筑物的边缘特征标签。第一遥感影像样本的分辨率小于低分辨率遥感影像的分辨率。
71.其中,第一遥感影像样本可以按照预设的超分倍率,通过ipg模块挖掘低分辨率遥感影像内部的自相关性,对低分辨率遥感影像做降采样处理,获得与低分辨率遥感影像采样尺度不同的第一遥感影像样本。
72.以常见的4倍超分辨率重建为例,如图5所示,将采样尺度为2m的低分辨率遥感影像i
low
(2m)裁切(crop&split)为若干小图,构成初始影像切片库。
73.然后,按照4倍的超分倍率构建初始影像切片库中每个小图对应的更小分辨率的影像,获得初始影像配对库。
74.其中,初始影像配对库中的各个影像的分辨率为8m,并且,初始影像配对库中的影像与初始影像切片库中影像一一对应。
75.初始影像切片库中的各个影像作为训练阶段的高分影像数据(hr training dataset i
lr
)。初始影像配对库中的各个影像即为训练阶段的低分影像数据(lr training dataset i
llr
)。由此训练获得的生成器可以由低分辨率遥感影像i
low
获得超分后的高分影像i
hr
。如图5所示,当i
low
为2m的分辨率时,可以利用生成器获得0.5m分辨率的i
hr

76.在一个示例中,在训练网络之前,还可以使用随机水平旋转、随机角度顺时针旋转、随机比例缩放、添加随机噪声、添加随机模糊、随机裁切等多个数据增强方法处理前述的初始影像切片库和初始影像配对库中的影像数据。
77.低分辨率遥感影像对应的建筑物的边缘特征标签可以通过人工对低分辨率遥感影像标注获得。
78.该步骤通过挖掘低分辨率影像内部的自相关性构建对抗网络的训练样本,而不引入大量外部数据,一方面可以减少计算量和获取遥感影像的成本,另一方面可以提高对抗网络对低分辨率影像超分重建的准确性。其中,对抗网络训练过程依赖于训练数据所提供的先验信息,而从大量外部数据中所获得的先验信息很可能并不适用于待处理的低分辨率
遥感影像。遥感影像内部在低尺度特征的自相关性,对于影像重建能提供更有价值的先验信息。
79.在步骤s402中,根据训练样本更新生成式对抗网络中生成器和判别器的参数。
80.在确定训练样本之后,对生成式对抗网络进行训练。在训练过程中,生成器不断学习如何重建出更真实的高分辨率的遥感图像以期迷惑鉴别器。而鉴别器则不断学习,努力判断出输入的影像是否为真实的高分辨率遥感影像。生成器和鉴别器相互博弈,从而使生成器可以重建出更高质量、更接近真实高分辨率遥感图像的超分辨率遥感影像的目的。
81.具体的训练过程包括如下步骤:
82.s4021.将指示第一遥感影像样本的初始影像配对库中的影像确定输入生成器中第一网络。其中,第一网络的各个层和模块对其进行特征编码和解码,输出超分辨率遥感影像(sr images)。超分辨率遥感影像即为输入的低分辨率遥感影像的预测结果。
83.s4022.将超分辨率遥感影像的梯度图输入生成器中第二网络。其中,第二网络的卷积层和激活层对其进行特征编码获得纹理信息,第二网络的frame branch纹理信息中提取出高频信息,第二网络的mask branch根据低分辨率遥感影像对应的建筑物的先验信息,在纹理信息中获得建筑物区域的信息,第二网络的解码层对frame branch获得的高频信息和mask branch获得的建筑物区域的信息进行融合解码,输出超分辨率遥感影像的增强后的梯度图(enhanced high frequency information),该增强后的梯度图指示低分辨率遥感影像对应的建筑物的预测的边缘特征。其中,两个分支输出的信息结合起来再解码而不是分别直接解码,可以有助于过滤掉冗余的高频信息。
84.s4023.将超分辨率遥感影像和超分辨率遥感影像的增强后的梯度图输入判别器,获得判别器的输出。如图3所示,判别器通过先提取出输入影像的低层到高层语义信息,经过一个全连接层dense(01024)重组后得到1024维特征,随后经由全连接层dense(1)获得一个一维数组,该数组即指征了判别器认为输入影像更趋向于真实的高分辨率遥感影像还是重建后的假的高分辨率遥感影像。
85.s4024.将第一网络的输出、第二网络的输出和判别器的输出代入损失函数,计算损失值,根据损失值更新生成器和判别器的参数。损失函数根据基于影像的像素级损失函数、感知损失函数(公式2所示的特征距离函数)、对抗损失函数(公式3所示)以及基于影像梯度的像素级损失函数(公式4所示)和对抗损失函数加权求和确定。
86.由于受到一范式损失函数的影响,第一网络为了获得更低的一范式损失得分,整体重建结果会在细节表现上倾向于更平滑,从而损失了我们需要的高频细节。因此,采用第二网络输出的边缘特征对第一网络输出的超分辨率遥感影像进行特征增强。如图3所示,将第一网络输出的影像中去除该影像的梯度图之后,添加第二网络输出的增强的高频信息(即建筑物的边缘特征,从而获得特征增强后的超分辨率遥感影像。
87.基于影像的像素级损失函数如公式(1)所示,该函数用于计算第一遥感影像样本的超分辨率遥感影像和高分辨率遥感影像之间像素的l1距离l
pix

[0088][0089]
公式(1)中,yi表示第一遥感影像样本的高分辨率遥感影像(即低分辨率遥感影像)中的第i个像素,表示第一遥感影像样本的超分辨率遥感影像中的第i个像素,
表示的范数。
[0090]
基于影像的感知损失函数如公式(2)所示,该函数用于测量第一遥感影像样本的超分辨率遥感影像和高分辨率遥感影像在特征空间的距离。其中,使用预先训练的19层vgg网络(用φ函数表示)提取两个影像的特征图,代入公式(2)计算损失值l
g_pix

[0091][0092]
基于影像的对抗损失函数如公式3所示,该函数鼓励生成器生成感官上更加真实的图像。
[0093][0094]
公式(3)中,表示判别器输出第一遥感影像样本的超分辨率遥感影像中的第i个像素的概率值。
[0095]
基于影像梯度的像素级损失函数如公式(4)所示,该函数用于计算第一遥感影像样本的超分辨率遥感影像和高分辨率遥感影像之间梯度的l1距离l
pix

[0096][0097]
公式(4)中,表示第一遥感影像样本的超分辨率遥感影像中的第i个像素的梯度,f(yi)表示第一遥感影像样本的高分辨率遥感影像中的第i个像素的梯度。
[0098]
基于影像梯度的对抗损失函数如公式(5)所示,该函数鼓励生成器生成感官上更加真实的梯度。
[0099][0100]
基于上述图2所示的建筑物影像的提取方法实施例,本技术实施例还提供一种提取模型的训练方法。
[0101]
参阅图2,该提取模型采用双路的结构,充分利用上一阶段生成的超分辨率遥感影像和增强的建筑物的边缘特征,以高效、准确的提取出高分辨率的建筑物影像。其中,一路用于对超分辨率遥感影像进行特征编码,从低层到高层逐层提取语义信息。另一路用于对建筑物边缘特征进行特征编码。由于本技术的目的是更好的提取建筑物,直观来说,建筑物的边缘特征对于重建建筑物的边缘和抑制建筑物周边的冗余信息具有重要意义。并且,单独设立并行的一路进行特征提取,而不是采用串行或其他方式,主要是为了更好的强化边缘特征对提取结果的引导作用。而经由双路并行的编码结构之后得到的两路编码特征,在维度和权重上具有相同的重要性,这对于进一步经由特征提取编码结果进行重建具有更好的引导价值。
[0102]
在编码结束后,提取模型中的融合层和解码层对两路的编码特征进行融合后进行解码,以获得建筑物的提取结果。其中,设计融合层主要是为了允许模型更好地对提取到的双路特征进行多层级的组合,从而让解码层可以轻易地从提取到的特征里重建出建筑物。
[0103]
图6是本技术实施例提供的一种提取模型的训练方法的流程图。该方法可以应用于上述的提取设备。如图6所示,该方法包括如下的步骤s601-s602。
[0104]
在步骤s601中,获取训练样本,训练样本包括:第二遥感影像样本、第二遥感影像样本对应的建筑物的边缘特征以及第二遥感影像样本对应的建筑物影像样本。
[0105]
其中,第二遥感影像样本对应的建筑物的边缘特征可以使用前述生成器中的第二网络处理第二遥感影像样本的梯度图获得。第二遥感影像样本对应的建筑物标签可以通过人工确定。
[0106]
在步骤s602中,根据训练样本更新提取模型的参数。
[0107]
将第二遥感影像样本和第二遥感影像样本对应的建筑物的边缘特征输入提取模型,在提取模型对经过特征编码、融合和解码之后,获得提取模型输出的建筑物影像。
[0108]
使用损失函数确定建筑物影像样本和提取模型输出的建筑物影像之间像素的l1距离,其中,损失函数如公式(6)所示。
[0109][0110]
公式(6)中,表示提取模型输出的建筑物影像中第i个像素,si表示建筑物影像样本中第i个像素。
[0111]
在一个示例中,可以对前述的各个损失函数加权求和,获得总损失函数,根据该总损失函数更新生成式对抗网络和提取模型的参数。
[0112]
基于上述图2所示的方法实施例,本技术实施例还提供一种跨分辨率的建筑物影像提取装置。该提取装置用于实现上述图2所述的方法实施例。如图7所示,该提取装置700包括:获取模块701、超分模块702和提取模块703。
[0113]
其中,获取模块701用于获取低分辨率遥感影像,所述低分辨率遥感影像中包含建筑物。
[0114]
其中,超分模块702用于利用生成式对抗网络中的生成器对所述低分辨率遥感影像进行超分辨率重建,确定超分辨率遥感影像和所述建筑物的边缘特征。
[0115]
其中,提取模块703用于根据所述边缘特征,利用提取模型在所述超分辨率遥感影像中提取所述建筑物的影像。
[0116]
需要说明的是,图7所示实施例提供的提取装置700在执行提取方法时,仅以上述各功能模块的划分举例说明,实际应用中,可以根据需要而将上述功能分配由不同的功能模块完成,即将设备的内部结构划分成不同的功能模块,以完成以上描述的全部或者部分功能。另外,上述实施例提供的提取装置与图2所示的提取方法实施例属于同一构思,其具体实现过程详见方法实施例,这里不再赘述。
[0117]
图8是本技术实施例提供一种提取设备800的硬件结构示意图。
[0118]
该提取设备800可以用于实现上述的建筑物的提取方法、生成式对抗网络的训练方法和提取模型的训练方法。参见图8,该提取设备800包括处理器820、存储器820、通信接口840和总线840,处理器820、存储器820和通信接口840通过总线840彼此连接。处理器820、存储器820和通信接口840也可以采用除了总线840之外的其他连接方式连接。
[0119]
其中,存储器820可以是各种类型的存储介质,例如随机存取存储器(random access memory,ram)、只读存储器(read-only memory,rom)、非易失性ram(non-volatile ram,nvram)、可编程rom(programmable rom,prom)、可擦除prom(erasable prom,eprom)、电可擦除prom(electrically erasable prom,eeprom)、闪存、光存储器、硬盘等。
[0120]
其中,处理器820可以是通用处理器,通用处理器可以是通过读取并执行存储器(例如存储器820)中存储的内容来执行特定步骤和/或操作的处理器。例如,通用处理器可以是中央处理器(central processing unit,cpu)。处理器820可以包括至少一个电路,以执行图2、图4或者图6所示实施例提供的方法的全部或部分步骤。
[0121]
其中,通信接口840包括输入/输出(input/output,i/o)接口、物理接口和逻辑接口等用于实现提取设备800内部的器件互连的接口,以及用于实现提取设备800与其他设备(例如其他提取设备或用户设备)互连的接口。物理接口可以是以太网接口,光纤接口,atm接口等。
[0122]
其中,总线840可以是任何类型的,用于实现处理器820、存储器820和通信接口840互连的通信总线,例如系统总线。
[0123]
上述器件可以分别设置在彼此独立的芯片上,也可以至少部分的或者全部的设置在同一块芯片上。将各个器件独立设置在不同的芯片上,还是整合设置在一个或者多个芯片上,往往取决于产品设计的需要。本技术实施例对上述器件的具体实现形式不做限定。
[0124]
图8所示的提取设备800仅仅是示例性的,在实现过程中,提取设备800还可以包括其他组件,本文不再一一列举。
[0125]
在上述实施例中,可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时,可以全部或部分地以计算机程序产品的形式实现。所述计算机程序产品包括一个或多个计算机指令。在计算机上加载和执行所述计算机程序指令时,全部或部分地产生按照本发明实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中,或者从一个计算机可读存储介质向另一个计算机可读存储介质传输,例如,所述计算机指令可以从一个网站站点、计算机、服务器或数据中心通过有线(例如同轴电缆、光纤、数字用户线(dsl))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、服务器或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存取的任何可用介质或者是包含一个或多个可用介质集成的服务器、数据中心等数据存储设备。所述可用介质可以是磁性介质(例如,软盘、硬盘、磁带)、光介质(例如,dvd)、或者半导体介质(例如,固态硬盘(solid state disk,ssd))等。
[0126]
可以理解的是,在本技术的实施例中涉及的各种数字编号仅为描述方便进行的区分,并不用来限制本技术的实施例的范围。应理解,在本技术实施例中,上述各过程的序号的大小并不意味着执行顺序的先后,各过程的执行顺序应以其功能和内在逻辑确定,不应对本技术实施例的实施过程构成任何限定。
[0127]
以上所述的具体实施方式,对本技术的目的、技术方案和有益效果进行了进一步详细说明,所应理解的是,以上所述仅为本发明的具体实施方式而已,并不用于限定本技术的保护范围,凡在本技术的技术方案的基础之上,所做的任何修改、等同替换、改进等,均应包括在本技术的保护范围之内。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1