发明内容
有鉴于此,本发明的目的在于提供一种高分辨率遥感影像半监督分类方法、装置、设备及介质,可以提高高分辨率遥感影像的分类精度。其具体方案如下:
一种高分辨率遥感影像半监督分类方法,包括:
对遥感影像进行预处理;
根据处理后的所述遥感影像,制作地物分类样本集;
构建基于Unet++网络的遥感影像语义分割模型,并通过所述地物分类样本集对所述语义分割模型进行训练;
构建基于近红外波段的阈值分割模型;
将所述语义分割模型和所述阈值分割模型进行模型融合,获取分类模型;
使用所述分类模型对待分类遥感影像进行分类。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述对遥感影像进行预处理,包括:
对遥感影像进行全色和多光谱影像融合、辐射校正、大气校正、几何校正的处理。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述根据处理后的所述遥感影像,制作地物分类样本集,包括:
建立与处理后的所述遥感影像相同大小的矢量,并将所述矢量全要素分割,分成水体、植被、裸土、不透水面四个类型;
对分成的类型进行标注,并根据类型字段不同,转换为栅格数据;
对处理后的所述遥感影像进行切片处理,同时裁剪出对应位置的标注的栅格,得到一组大小相同的图片和标签作为地物分类样本集。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述Unet++网络采用多级上采样和跳层连接的方法,提取多层特征。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述Unet++网络包括下采样层,上采样层,以及用于对所述下采样层进行特征抽取的中间层;其中,
所述下采样层添加有EfficientB4模型的特征提取部分。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述构建基于近红外波段的阈值分割模型,包括:
使用阈值分割法获取处理后的所述遥感影像的近红外波段阈值直方图;
将所述地物分类样本集和所述近红外波段阈值直方图进行匹配,选取出所述近红外波段阈值直方图中水体和植被的阈值区域,以构建阈值分割模型。
优选地,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,所述将所述语义分割模型和所述阈值分割模型进行模型融合,获取分类模型,包括:
当所述语义分割模型的输出结果与所述阈值分割模型的输出结果均为水体或植被时,或,当所述语义分割模型的输出结果为裸土或不透水面且所述阈值分割模型的输出结果为其他时,判定分类结果正确;
将判定为正确的分类结果作为新的地物分类样本集,使用迁移学习的方法继续训练所述语义分割模型,获取分类模型。
本发明实施例还提供了一种高分辨率遥感影像半监督分类装置,包括:
影像处理模块,用于对遥感影像进行预处理;
样本集制作模块,用于根据处理后的所述遥感影像,制作地物分类样本集;
第一模型构建模块,用于构建基于Unet++网络的遥感影像语义分割模型,并通过所述地物分类样本集对所述语义分割模型进行训练;
第二模型构建模块,用于构建基于近红外波段的阈值分割模型;
模型融合模块,用于将所述语义分割模型和所述阈值分割模型进行模型融合,获取分类模型;
影像分类模块,用于使用所述分类模型对待分类遥感影像进行分类。
本发明实施例还提供了一种高分辨率遥感影像半监督分类设备,包括处理器和存储器,其中,所述处理器执行所述存储器中存储的计算机程序时实现如本发明实施例提供的上述高分辨率遥感影像半监督分类方法。
本发明实施例还提供了一种计算机可读存储介质,用于存储计算机程序,其中,所述计算机程序被处理器执行时实现如本发明实施例提供的上述高分辨率遥感影像半监督分类方法。
从上述技术方案可以看出,本发明所提供的一种高分辨率遥感影像半监督分类方法,包括:对遥感影像进行预处理;根据处理后的遥感影像,制作地物分类样本集;构建基于Unet++网络的遥感影像语义分割模型,并通过地物分类样本集对语义分割模型进行训练;构建基于近红外波段的阈值分割模型;将语义分割模型和阈值分割模型进行模型融合,获取分类模型;使用分类模型对待分类遥感影像进行分类。
本发明通过构建基于Unet++网络的遥感影像语义分割模型和基于近红外波段的阈值分割模型,然后使用多模型融合的方法,使遥感影像的纹理信息和近红外波段的光谱信息融合,再对高分辨率遥感影像分类,能够提高分类精度。
此外,本发明还针对高分辨率遥感影像半监督分类方法提供了相应的装置、设备及计算机可读存储介质,进一步使得上述方法更具有实用性,该装置、设备及计算机可读存储介质具有相应的优点。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
本发明提供一种高分辨率遥感影像半监督分类方法,如图1和图2所示,包括以下步骤:
S101、对遥感影像进行预处理;
S102、根据处理后的遥感影像,制作地物分类样本集;
S103、构建基于Unet++网络的遥感影像语义分割模型,并通过地物分类样本集对语义分割模型进行训练;
S104、构建基于近红外波段的阈值分割模型;
S105、将语义分割模型和阈值分割模型进行模型融合,获取分类模型;
S106、使用分类模型对待分类遥感影像进行分类。
在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,通过构建基于Unet++网络的遥感影像语义分割模型和基于近红外波段的阈值分割模型,然后使用多模型融合的方法,使遥感影像的纹理信息和近红外波段的光谱信息融合,再对高分辨率遥感影像分类,能够提高分类精度。
进一步地,在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,步骤S101对遥感影像进行预处理,具体可以包括:对遥感影像进行全色和多光谱影像融合、辐射校正、大气校正、几何校正等处理,获取一个具有四波段(RGB和近红外)的高分辨率遥感影像。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,由于遥感影像尺寸过大,需要裁剪为切片,赋予属性,步骤S102根据处理后的遥感影像,制作地物分类样本集,具体可以包括:
首先,建立与处理后的遥感影像相同大小的矢量,并将矢量全要素分割,分成水体、植被、裸土、不透水面四个类型;
然后,对分成的类型进行标注,并根据类型字段不同,转换为栅格数据;具体可以使用shpfile文件进行标注,然后将shpfile根据类型字段不同,转换为栅格数据;
最后,对处理后的遥感影像进行切片处理;具体地,将遥感影像切成一个个小的切片便于训练,切片大小可以设置为512×512,切片之间不重叠;同时裁剪出对应位置的标注的栅格,得到一组大小相同的图片和标签作为地物分类样本集。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,如图3所示,Unet++网络采用多级上采样和跳层连接的方法,提取多层特征。需要说明的是,Unet网络是语义分割中常用的一种分割网络模型,首先通过卷积进行下采样,提取一层又一层的特征,然后进行上采样,并将下采样和上采样之间的特征连接,由于这个模型结构图形状类似U形,所以名为Unet,其中下采样的过程是一个编码的过程,上采样是解码的过程;而Unet++网络是在Unet的基础上进行了增加,采用了多级上采样和跳层连接的方法,提取更多层的特征。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,Unet++网络具体可以包括下采样层,上采样层,以及用于对下采样层进行特征抽取的中间层;其中,下采样层添加有EfficientB4模型的特征提取部分。也就是说,本发明在Unet的基础上,将EfficientB4模型的特征提取部分加入到Unet++网络的编码过程中,改进了网络结构,提取了更多的特征。
具体地,步骤S103构建基于Unet++网络的遥感影像语义分割模型,具体可以包括以下步骤:
首先,定义一个卷积block,如图4所示,一个卷积block包括一个卷积层、BatchNormalization(BN)层、一个LeakyRelU激活函数;
然后,定义一个残差block,如图5所示,将残差block网络的输入经过LeakReLU和BN层之后,再经过上面定义的两个卷积block,之后加上原始输入经过BN层处理之后的结果;
之后,建立Unet++网络的下采样层、上采用层和中间层。下采样层是特征提取的部分,与Unet网络相同,这里使用四个下采样层作为Unet++网络的下采样层,不同的是,为了更深入地提出特征,该网络的下采样层是从EfficientB4获取,即分别从EfficientB4的第342、154、92、30层作为Unet++中下采样的四个层conv4、conv3、conv2、conv1;中间层是对下采样层的进一步特征抽取,不予不同层级的特征,抽取次数也不一样。对于conv4,编码为deconv4,再抽取三次,获得三级特征层deconv4_up1、deconv4_up2、deconv4_up3,然后将conv4提取特征为deconv3,并提取特征deconv3_up1,deconv3_up2,将deconv3、conv3与deconv4_up1相加,获得uconv3,将uconv3编码为deconv2,并提取特征deconv2_up1,然后将deconv2、conv2、deconv4_up2、deconv3_up1相加,得到uconv2,将uconv2编码为deconv1,然后将conv1、deconv1、deconv2_up1、deconv3_up2、deconv4_up3相加,获得uconv1,并编码为uconv0,最后将uconv0做一个卷积,将特征降到1维,作为网络输出;
最后,定义损失函数,使用网络预测结果与真实标签之间的dice loss作为损失函数。使用余弦退火的方法设定learning rate(lr),初始lr设置为0.001,epoch设置为30,batchsize设置为32,然后开始训练模型。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,步骤S104构建基于近红外波段的阈值分割模型,具体可以包括:使用阈值分割法获取处理后的遥感影像的近红外波段阈值直方图;将地物分类样本集和近红外波段阈值直方图进行匹配,选取出近红外波段阈值直方图中水体和植被的阈值区域,以构建阈值分割模型。
具体地,考虑到近红外波段对水体的吸收较强,而且植被中含有较多的水体,所以水体和植被的反射率较低,使用波段四来判断水体和植被。步骤S104中使用阈值分割的方法,只提取高分辨率遥感影像中的波段四,显示直方图,对于8位影像,直方图横坐标范围为0到255,如图6所示,将训练样本和直方图进行匹配,选取直方图中水体聚集的一段作为影像中水体的反射率部分,通常靠左,记录水体的阈值区域,同样,选取植被的阈值区域。待分类的数据在水体或者植被的阈值内时,标记为水体或植被。这样充分利用近红外波段的特征,能够一定程度上增加特定目标的分类精度。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,步骤S105将语义分割模型和阈值分割模型进行模型融合,获取分类模型,具体可以包括:当语义分割模型的输出结果与阈值分割模型的输出结果均为水体或植被时,或,当语义分割模型的输出结果为裸土或不透水面且阈值分割模型的输出结果为其他时,判定分类结果正确;将判定为正确的分类结果作为新的地物分类样本集,使用迁移学习的方法继续训练语义分割模型,获取分类模型。这样使用语义分割对遥感影像进行分类,同时针对部分地物类型如水体、植被等,使用近红外波段特征进行提取,对两个结果进行决策级融合,然后使用半监督分类的方法增加样本,并再次分类,可以获取更高的分类精度。
具体地,步骤S105使用决策级融合的方法,语义分割模型的输出结果是一个二维矩阵,每个位置代表该像元的类别,总共包含四个类别,阈值分割的结果也是一个二维矩阵,每个位置代表该像元的类别,与语义分割不同的是,这里只包含三个类别,水体、植被、其它。对于每一个像素,当语义分割获取结果为裸土或不透水面,阈值分割获取为其他,或者都为水体或者植被,则认为分类正确。当分类结果不同时,对这部分不分类,将已经分类的作为新的训练样本,使用迁移学习的方法,继续训练语义分割模型,获取最终的分类结果。
在具体实施时,在本发明实施例提供的上述高分辨率遥感影像半监督分类方法中,步骤S106使用分类模型对待分类遥感影像进行分类,具体可以包括:首先,将待分类遥感影像切片,切片大小与步骤一相同,然后将每一个切片输入到分类模型中分类,经过膨胀、腐蚀等后处理操作,再把最后各个切片的分类结果拼接起来,添加坐标信息,生成tiff文件,实例如图7所示。
需要说明的是,本发明实施例提供的上述高分辨率遥感影像半监督分类方法主要是考虑高分辨率遥感影像的纹理特征和近红外波段的物理特征,通过改进的Unet++语义分割网络与阈值分割相结合的方式,采用半监督分类的方法,扩充训练样本,并再次训练语义分割模型以获取更高的分类精度,最后经过分类后处理,生成分类结果。
基于同一发明构思,本发明实施例还提供了一种高分辨率遥感影像半监督分类装置,由于该装置解决问题的原理与前述一种高分辨率遥感影像半监督分类方法相似,因此该装置的实施可以参见高分辨率遥感影像半监督分类方法的实施,重复之处不再赘述。
在具体实施时,本发明实施例提供的高分辨率遥感影像半监督分类装置,如图8所示,具体包括:
影像处理模块11,用于对遥感影像进行预处理;
样本集制作模块12,用于根据处理后的遥感影像,制作地物分类样本集;
第一模型构建模块13,用于构建基于Unet++网络的遥感影像语义分割模型,并通过地物分类样本集对语义分割模型进行训练;
第二模型构建模块14,用于构建基于近红外波段的阈值分割模型;
模型融合模块15,用于将语义分割模型和阈值分割模型进行模型融合,获取分类模型;
影像分类模块16,用于使用分类模型对待分类遥感影像进行分类。
在本发明实施例提供的上述高分辨率遥感影像半监督分类装置中,可以通过上述六个模块的相互作用,使遥感影像的纹理信息和近红外波段的光谱信息融合后,再对高分辨率遥感影像分类,提高了分类精度。
关于上述各个模块更加具体的工作过程可以参考前述实施例公开的相应内容,在此不再进行赘述。
相应地,本发明实施例还公开了一种高分辨率遥感影像半监督分类设备,包括处理器和存储器;其中,处理器执行存储器中存储的计算机程序时实现前述实施例公开的高分辨率遥感影像半监督分类方法。
关于上述方法更加具体的过程可以参考前述实施例中公开的相应内容,在此不再进行赘述。
进一步地,本发明还公开了一种计算机可读存储介质,用于存储计算机程序;计算机程序被处理器执行时实现前述公开的高分辨率遥感影像半监督分类方法。
关于上述方法更加具体的过程可以参考前述实施例中公开的相应内容,在此不再进行赘述。
本说明书中各个实施例采用递进的方式描述,每个实施例重点说明的都是与其它实施例的不同之处,各个实施例之间相同或相似部分互相参见即可。对于实施例公开的装置、设备、存储介质而言,由于其与实施例公开的方法相对应,所以描述的比较简单,相关之处参见方法部分说明即可。
专业人员还可以进一步意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、计算机软件或者二者的结合来实现,为了清楚地说明硬件和软件的可互换性,在上述说明中已经按照功能一般性地描述了各示例的组成及步骤。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本申请的范围。
结合本文中所公开的实施例描述的方法或算法的步骤可以直接用硬件、处理器执行的软件模块,或者二者的结合来实施。软件模块可以置于随机存储器(RAM)、内存、只读存储器(ROM)、电可编程ROM、电可擦除可编程ROM、寄存器、硬盘、可移动磁盘、CD-ROM、或技术领域内所公知的任意其它形式的存储介质中。
综上,本发明实施例提供的一种高分辨率遥感影像半监督分类方法,包括:对遥感影像进行预处理;根据处理后的遥感影像,制作地物分类样本集;构建基于Unet++网络的遥感影像语义分割模型,并通过地物分类样本集对语义分割模型进行训练;构建基于近红外波段的阈值分割模型;将语义分割模型和阈值分割模型进行模型融合,获取分类模型;使用分类模型对待分类遥感影像进行分类。这样通过构建基于Unet++网络的遥感影像语义分割模型和基于近红外波段的阈值分割模型,然后使用多模型融合的方法,使遥感影像的纹理信息和近红外波段的光谱信息融合,再对高分辨率遥感影像分类,能够提高分类精度。此外,本发明还针对高分辨率遥感影像半监督分类方法提供了相应的装置、设备及计算机可读存储介质,进一步使得上述方法更具有实用性,该装置、设备及计算机可读存储介质具有相应的优点。
最后,还需要说明的是,在本文中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下,由语句“包括一个……”限定的要素,并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。
以上对本发明所提供的高分辨率遥感影像半监督分类方法、装置、设备及介质进行了详细介绍,本文中应用了具体个例对本发明的原理及实施方式进行了阐述,以上实施例的说明只是用于帮助理解本发明的方法及其核心思想;同时,对于本领域的一般技术人员,依据本发明的思想,在具体实施方式及应用范围上均会有改变之处,综上所述,本说明书内容不应理解为对本发明的限制。