[发明专利]多光谱图像的分割方法、装置、电子设备及存储介质

说明书

本申请提供了一种多光谱图像的分割方法、装置、电子设备、计算机可读存储介质；方法包括：获取待分割的多光谱图像，并对多光谱图像进行波段组合，得到至少两个波段组合图像；通过图像分割模型中第一分割层包括的至少两个分割子模型，分别对各波段组合图像进行图像分割，得到相应的分割特征图像；其中，分割子模型与波段组合图像一一对应；通过图像分割模型的特征拼接层，对各分割特征图像进行特征拼接，得到特征图像；通过图像分割模型的第二分割层，对特征图像进行图像分割，得到与多光谱图像对应的目标分割图像。通过本申请，能够提高多光谱图像的分割精度。

技术领域

本申请涉及计算机视觉技术，尤其涉及一种多光谱图像的分割方法、装置、电子设备及计算机可读存储介质。

背景技术

随着遥感传感器等对地观测技术的迅速发展，多光谱图像的数据类型变得越来越丰富，图像空间分辨率也越来越高，高分辨率图像拥有丰富且精细的地物信息，地物的细节也更加清晰，为多光谱图像中地物信息的提取提供了良好的研究基础，但同时也导致地物的结构比较复杂，干扰信息较难处理。

相关技术中，针对多光谱图像的分割方法，受限于复杂的背景、影像质量、影像分辨率、季节和天气因素、地物尺度差异、样本不平衡等，在进行多光谱图像分割操作时，只考虑到多光谱图像的浅层特征，未能充分利用多光谱图像中不同波段的波段信息，使得多光谱图像分割精度偏低。

发明内容

本申请实施例提供一种多光谱图像的分割方法、装置、电子设备、计算机可读存储介质及计算机程序产品，能够提高多光谱图像的分割精度。

本申请实施例的技术方案是这样实现的：

本申请实施例提供一种多光谱图像的分割方法，包括：

获取待分割的多光谱图像，并对所述多光谱图像进行波段组合，得到至少两个波段组合图像；

通过图像分割模型中第一分割层包括的至少两个分割子模型，分别对各所述波段组合图像进行图像分割，得到相应的分割特征图像；

其中，所述分割子模型与所述波段组合图像一一对应；

通过所述图像分割模型的特征拼接层，对各所述分割特征图像进行特征拼接，得到特征图像；

通过所述图像分割模型的第二分割层，对所述特征图像进行图像分割，得到与所述多光谱图像对应的目标分割图像。

本申请实施例提供一种多光谱图像的分割装置，包括：

图像预处理模块，用于获取待分割的多光谱图像，并对所述多光谱图像进行波段组合，得到至少两个波段组合图像；

第一分割模块，用于通过图像分割模型中第一分割层包括的至少两个分割子模型，分别对各所述波段组合图像进行图像分割，得到相应的分割特征图像；其中，所述分割子模型与所述波段组合图像一一对应；

特征拼接模块，用于通过所述图像分割模型的特征拼接层，对各所述分割特征图像进行特征拼接，得到特征图像；

第二分割模块，用于通过所述图像分割模型的第二分割层，对所述特征图像进行图像分割，得到与所述多光谱图像对应的目标分割图像。

上述方案中，所述图像预处理模块，还用于对所述多光谱图像进行波段提取，得到多个波段中各波段对应的单波段图像；

对多个所述单波段图像进行波段组合，得到至少两个波段组合图像。

上述方案中，所述图像预处理模块，还用于获取所述多光谱图像的图像内容所归属的类别，以及各所述波段对应的波段参数；

基于所述类别以及所述波段参数，从多个所述单波段图像中选择单波段图像形成至少两个单波段图像组，每个所述单波段图像组包括至少两个所述单波段图像；