一种融合遥感图像与点云数据的地理要素识别方法转让专利

申请号 : CN202310539805.2

文献号 : CN116258970B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 赵薛强陈洋波刘俊孙怀张

申请人 : 中山大学

摘要 :

本发明提供一种融合遥感图像与点云数据的地理要素识别方法,涉及图像数据处理的技术领域,首先采集图像数据和点云数据并进行类别标注,构建为样本数据集,再构建融合图像与点云数据的语义分割模型,提取图像数据的2D特征和点云数据的3D特征并融合得到图像和点云的分割预测值,利用样本数据集训练融合图像与点云数据的语义分割模型,最后将待识别遥感图像、待识别遥感图像对应的点云数据输入训练好的融合图像与点云数据的语义分割模型,输出图像分割结果和点云分割结果,利用不同的色彩对不同类别的点云分割结果进行标记,实现地理要素的识别,充分利用了2D图像信息的有序性以及3D点云数据的空间性,提高了地理要素的分割识别精度。

权利要求 :

1.一种融合遥感图像与点云数据的地理要素识别方法,其特征在于,包括:S1.采集图像数据和点云数据,对图像数据和点云数据分别进行类别标注,分别作为图像数据和点云数据的真实标签,将带有类别标注的图像数据和点云数据构建为样本数据集,将样本数据集划分为训练集、验证集、测试集;

S2.构建融合图像与点云数据的语义分割模型,融合图像与点云数据的语义分割模型包括用于提取图像数据的2D特征的2D分支2D Branch、用于提取点云数据的3D特征的3D分支3D Branch及融合结构Feature Fusion,所述2D分支2D Branch包括2D图像分割模块Fast 

2D

SCNN,输出2D特征E ;所述3D分支3D Branch包括3D点云数据分割模块PFTransformer,输出

3D

3D特征E ;融合结构Feature Fusion连接2D分支和3D分支,将2D特征和3D特征融合,过程如下:

3D

3D点云数据分割模块PFTransformer输出的3D特征E 经过第一共享感知机Shared 

3D′ 3D′ 2D

MLP1后,输出为3D共享特征E ,令E 与2D图像分割模块FastSCNN输出的2D特征E 通过Concat函数进行拼接,将拼接后的特征通过第二共享感知机Shared MLP2进行特征融合,得

2D3D

到融合特征E ;

2D3D

将融合特征E 通过第三共享感知机Shared MLP3后进行Sigmoid函数计算,得到注意

2D3D 3D

力权重,注意力权重与融合特征E 进行Hadamard乘积,再以残差的方式与2D特征E 按位置相加,并通过共享感知机Shared MLP组成的2D分类器2D Classfier,获得2D特征的图像语义分割预测值;

3D

3D点云数据分割模块PFTransformer输出的3D特征E 与经过第一共享感知机Shared 

3D′

MLP1的得到的3D共享特征E 进行融合,再通过3D分支3D  Branch的3D分类器3D Classfier,获得3D点云的点云语义分割预测值;利用2D特征优化3D特征,得到图像分割预测值和点云分割预测值;

S3.利用训练集对融合图像与点云数据的语义分割模型进行训练,训练过程中分别计算图像分割预测值和点云分割预测值与真实标签图之间的损失值,根据损失值调整融合图像与点云数据的语义分割模型的参数,直到融合图像与点云数据的语义分割模型收敛,然后利用验证集对训练过程中的融合图像与点云数据的语义分割模型进行评估,得出训练好的融合图像与点云数据的语义分割模型,并利用测试集测试融合图像与点云数据的语义分割模型的预测准确度;

S4.将待识别遥感图像、待识别遥感图像对应的点云数据输入训练好的融合图像与点云数据的语义分割模型,输出图像分割结果和点云分割结果;

S5.利用不同的色彩对不同类别的点云分割结果进行标记,展示分割结果,实现地理要素的识别。

2.根据权利要求1所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述2D图像分割模块FastSCNN包括依次连接的主干网络特征编码器、双分支全局特征提取器、特征融合器;

所述双分支全局特征提取器包括两个分支:第一个分支由Bottleneck卷积块、SE注意力模块、Pyramid Pooling池化组成,提取图像数据的全局特征;第二个分支保留图像数据的原始特征信息,两个分支分别与特征融合器连接;特征融合器将全局特征提取器的两个

2D

分支的输出特征相加融合并调整通道数量,输出2D特征E 。

3.根据权利要求2所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述3D点云数据分割模块PFTransformer包括3D点云特征编码器Encoding和3D点云特征解码器Decoding;

所述3D点云特征编码器Encoding包括依次连接的非下采样特征编码块和四层下采样特征编码块,输出编码后的点云特征;所述3D点云特征解码器Decoding包括依次连接的非

3D

上采样特征解码块和四层上采样特征解码块,对点云特征解码后得到3D特征E 。

4.根据权利要求3所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述2D分支2D Branch还包括2D分类器2D Classfier,用于在融合结构Feature Fusion对

2D特征和3D特征融合后进行分类,输出2D点云的图像语义分割预测值;

所述3D分支3D Branch还包括3D分类器3D Classfier,用于在融合结构Feature Fusion将2D特征和3D特征融合后进行分类,输出3D点云的点云语义分割预测值。

5.根据权利要求3所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述非下采样特征编码块包括共享感知机Shared MLP和特征自注意力块PFTransformer Block,每一层下采样特征编码块包括下采样结构Transition Down和特征自注意力块PFTransformer Block;

所述非上采样特征解码块包括共享感知机Shared  MLP和特征自注意力块PFTransformer Block,每一层上采样特征解码块包括上采样结构Transition UP和特征自注意力块PFTransformer Block。

6.根据权利要求5所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述特征自注意力块PFTransformer Block由n组循环结构构成,每组循环结构包含上层Shared MLP、位置与特征自注意力层PFTransformerLayer和下层Shared MLP。

7.根据权利要求5所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述下采样结构Transition Down基于点云特征的空间位置进行最远点采样,然后通过K近邻算法对采样点进行分组,利用共享感知机Shared MLP对每个点的组内特征进行特征提取,通过局部最大池化提取组内最大特征点,得到下采样特征;

所述上采样结构Transition UP利用共享感知机Shared MLP对输入的点云特征维度进行降维,然后根据点云特征的空间位置,使用三线性插值的方法对点云特征进行上采样,得到上采样特征。

8.根据权利要求6所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,所述位置与特征自注意力层PFTransformerLayer用于提取点云特征,过程如下:基于欧式空间位置的自注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ψp、αp,基于欧式空间距离对N个点云特征进行提取并分别记作qp、kp、N×Dvp,qp、kp、vp∈R ,其中,N是点云的数量,D是点云的特征维度;使用共享感知机Shared MLP对位置特征进行编码后获得特征δp;最后计算特征 的计算表达公式如下:其中,γp、ρp均表示共享感知机Shared MLP,即:共享感知机γp和共享感知机ρp,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点,⊙表示Hadamard乘积;

基于特征空间的子注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ψf、αf,基于欧式空间距离对N个点云特征进行提取并分别记作qf、kf、vf,N×Dqf、kf、vf∈R ;使用Shared MLP对提取的点云特征进行编码后获得特征δf;最后计算特征的计算表达公式如下:其中,γf、ρf均表示共享感知机Shared MLP,即:共享感知机γf和共享感知机ρf,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点,⊙表示Hadamard乘积;

使用Shared MLP将特征 融合,获得提取后的点云特征yi,yi的计算公式如下:其中,f表示语义特征空间的特征提取、p表示欧式空间的特征提取。

9.根据权利要求1所述的融合遥感图像与点云数据的地理要素识别方法,其特征在于,在步骤S4中,对2D图像分割模块FastSCNN进行训练,过程如下:将训练集中的图像数据输入2D图像分割模块FastSCNN,输出图像语义分割预测值Z2D,采用交叉熵损失函数计算预测值与样本图片特征真值GtI之间的损失Loss2D,计算公式如下:其中,B为输入2D图像分割模块FastSCNN的图像的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别,log表示对数函数;

对3D点云数据分割模块进行训练,过程如下:

将训练集中的图像数据输入3D点云数据分割模块,输出点云语义分割预测值Z3D,采用交叉熵损失函数计算预测值与点云特征真值GtP之间的损失Loss3D,计算公式如下:其中,B为输入3D点云数据分割模块的数据的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别,log表示对数函数;

训练3D点云数据分割模块,令3D点云数据分割模块输出的预测结果分布学习2D图像分割模块FastSCNN的预测结果分布,具体过程如下:利用知识蒸馏的方法,采用KL散度计算2D图像分割模块FastSCNN的预测结果分布与3D点云数据分割模块的预测结果分布的损失LossKL,计算公式如下:其中,M表示进行点云语义分割类别数量,ci表示分布中具体某一类别,U表示2D图像分割模块FastSCNN的预测结果分布,V表示3D点云数据分割模块的预测结果分布,log表示对数函数;

构建融合图像与点云数据的语义分割模型的总损失函数Losstotal,Losstotal的表达式如下:Losstotal=Loss2D(Z2D,GtI)+Loss3D(Z3D,GtP)+LossKL(U,V)最后,对总损失函数Losstotal进行优化,使损失值达到最小。

说明书 :

一种融合遥感图像与点云数据的地理要素识别方法

技术领域

[0001] 本发明涉及地理图像信息处理的技术领域,更具体地,涉及一种融合遥感图像与点云数据的地理要素识别方法。

背景技术

[0002] 随着遥感技术、深度学习技术的快速发展,对地理要素的智能识别一般采用图像的语义分割算法,利用图像分割算法提取图像中的语义信息,对不同的地理要素进行标记分类,达到识别的目的。但由于图像数据缺少3D空间信息,基于图像的语义分割算法,在地理要素识别中无法识别地理要素的高度、坡度等空间信息,导致地理要素分割不完整。
[0003] 随着遥感技术、深度学习技术的快速发展,为获得空间距离信息,可采用激光雷达扫描地形地貌,获得3D点云数据,对3D点云数据采用分割算法实现地理要素的识别,常用的点云分割算法有PointNet、PointNet++、PointTransformer等算法。使用点云数据的分割算法能够较好地区分地形高低、坡度,获得丰富的3D空间信息,但3D点云数据是无序的,缺少空间上的联系,使得分割算法收敛困难,分割精度较低。
[0004] 现有技术公开了一种地图要素生成方法,结合了图像数据与视觉三维重建技术,通过图像获得地图要素的形状点的坐标,再通过视觉三维重建得到形状点的初始地理坐标,最后通过激光点拟合函数表达式,确定地图要素的位置点地理坐标。然而,该方案虽然结合了图像数据和三维重建技术,但无法充分融合2D图像信息的有序性以及3D信息的空间性的优势,识别精度较低。

发明内容

[0005] 为解决当前地理要素识别方法无法充分利用2D图像信息的有序性以及3D点云数据的空间性,导致地理要素的分割识别精度较低的问题,本发明提供一种融合遥感图像与点云数据的地理要素识别方法,利用二维图像的有序性和三维点云的空间性,训练阶段通过投影方法将三维点云投影到对应的二维图像中,进行知识蒸馏学习,采用了多数据源融合的方式,提高了地理要素的分割识别精度。
[0006] 为了达到上述技术效果,本发明的技术方案如下:
[0007] 一种融合遥感图像与点云数据的地理要素识别方法,包括:
[0008] S1.采集图像数据和点云数据,对图像数据和点云数据分别进行类别标注,分别作为图像数据和点云数据的真实标签,将带有类别标注的图像数据和点云数据构建为样本数据集,将样本数据集划分为训练集、验证集、测试集;
[0009] S2.构建融合图像与点云数据的语义分割模型,融合图像与点云数据的语义分割模型包括用于提取图像数据的2D特征的2D分支2D Branch、用于提取点云数据的3D特征的3D分支3D Branch及融合结构Feature Fusion,2D分支2D Branch的输出端及3D分支3D Branch的输出端均连接融合结构Feature Fusion,融合结构Feature Fusion将2D特征和3D特征融合,利用2D特征优化3D特征,得到图像分割预测值和点云分割预测值;
[0010] S3.利用训练集对融合图像与点云数据的语义分割模型进行训练,训练过程中分别计算图像分割预测值和点云分割预测值与真实标签图之间的损失值,根据损失值调整融合图像与点云数据的语义分割模型的参数,直到融合图像与点云数据的语义分割模型收敛,然后利用验证集对训练过程中的融合图像与点云数据的语义分割模型进行评估,得出训练好的融合图像与点云数据的语义分割模型,并利用测试集测试融合图像与点云数据的语义分割模型的预测准确度;
[0011] S4.将待识别遥感图像、待识别遥感图像对应的点云数据输入训练好的融合图像与点云数据的语义分割模型,输出图像分割结果和点云分割结果;
[0012] S5.利用不同的色彩对不同类别的点云分割结果进行标记,展示分割结果,实现地理要素的识别。
[0013] 本技术方案使用多数据源融合的方法,将2D图像分割结果用于辅助3D点云分割,提高了地理要素的分割识别精度。
[0014] 优选地,所述2D分支2D Branch包括2D图像分割模块Fast SCNN,所述2D图像分割模块FastSCNN包括依次连接的主干网络特征编码器、双分支全局特征提取器、特征融合器;
[0015] 所述双分支全局特征提取器包括两个分支:第一个分支由Bottleneck卷积块、SE注意力模块、Pyramid Pooling池化组成,提取图像数据的全局特征;第二个分支保留图像数据的原始特征信息,两个分支分别与特征融合器连接;特征融合器将全局特征提取器的两个分支的输出特征相加融合并调整通道数量,输出2D特征 。
[0016] 优选地,所述3D分支3D Branch包括3D点云数据分割模块PFTransformer;所述3D点云数据分割模块PFTransformer包括3D点云特征编码器Encoding和3D点云特征解码器Decoding;
[0017] 所述3D点云特征编码器Encoding包括依次连接的非下采样特征编码块和四层下采样特征编码块,输出编码后的点云特征;所述3D点云特征解码器Decoding包括依次连接的非上采样特征解码块和四层上采样特征解码块,对点云特征解码后得到3D特征 。
[0018] 优选地,所述2D分支2D Branch还包括2D分类器2D Classfier,用于在融合结构Feature Fusion对2D特征和3D特征融合后进行分类,输出2D点云的图像语义分割预测值;
[0019] 所述3D分支3D Branch还包括3D分类器3D Classfier,用于在融合结构Feature Fusion将2D特征和3D特征融合后进行分类,输出3D点云的点云语义分割预测值。
[0020] 优选地,所述的融合结构Feature Fusion将2D特征和3D特征融合,其特征在于,过程如下:
[0021] 3D点云数据分割模块PFTransformer输出的3D特征 经过第一共享感知机Shared MLP1后,输出为3D共享特征 ,令 与2D图像分割模块FastSCNN输出的2D特征通过Concat函数进行拼接,将拼接后的特征通过第二共享感知机Shared MLP2进行特征融合,得到融合特征 ;
[0022] 将融合特征 通过第三共享感知机Shared MLP3后进行Sigmoid函数计算,得到注意力权重,注意力权重与融合特征 进行Hadamard乘积,再以残差的方式与2D特征按位置相加,并通过共享感知机Shared MLP组成的2D分类器2D Classfier,获得2D特征的图像语义分割预测值;
[0023] 3D点云数据分割模块PFTransformer输出的3D特征 与经过第一共享感知机Shared MLP1的得到的3D共享特征 进行融合,再通过3D分支3D Branch的3D分类器3D Classfier,获得3D点云的点云语义分割预测值。
[0024] 在此,通过知识蒸馏的方法,令3D点云数据分割模块PFTransformer学习2D图像分割模块FastSCNN的预测分布,间接使用图像数据的2D特征来优化点云数据的3D特征,让3D点云数据分割模块PFTransformer学习到图像中的更多信息。
[0025] 优选地,所述非下采样特征编码块包括共享感知机Shared MLP和特征自注意力块PFTransformer Block,每一层下采样特征编码块包括下采样结构Transition Down和特征自注意力块PFTransformer Block;
[0026] 所述非上采样特征解码块包括共享感知机Shared  MLP和特征自注意力块PFTransformer Block,每一层上采样特征解码块包括上采样结构Transition UP和特征自注意力块PFTransformer Block。
[0027] 优选地,所述特征自注意力块PFTransformer Block由n组循环结构构成,每组循环结构包含上层Shared MLP、位置与特征自注意力层PFTransformerLayer和下层Shared MLP。
[0028] 优选地,所述下采样结构Transition Down基于点云特征的空间位置进行最远点采样,然后通过K近邻算法对采样点进行分组,利用共享感知机Shared MLP对每个点的组内特征进行特征提取,通过局部最大池化提取组内最大特征点,得到下采样特征;
[0029] 所述上采样结构Transition UP利用共享感知机Shared MLP对输入的点云特征维度进行降维,然后根据点云特征的空间位置,使用三线性插值的方法对点云特征进行上采样,得到上采样特征。
[0030] 优选地,所述位置与特征自注意力层PFTransformerLayer用于提取点云特征,过程如下:
[0031] 基于欧式空间位置的自注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ,基于欧式空间距离对N个点云特征进行提取并分别记作 ,其中,N是点云的数量,D是点云的特征维度;使用共享感知机Shared MLP对位置特征进行编码后获得特征 ;最后计算特征 , 的计算表达公式如下:
[0032]
[0033] 其中, 、 表示共享感知机Shared MLP,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点, 表示Hadamard乘积;
[0034] 基于特征空间的子注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ,基于欧式空间距离对N个点云特征进行提取并分别记作;使用Shared MLP对提取的点云特征进行编码后获得特征;最后计算特征 ,的计算表达公式如下:
[0035]
[0036] 其中,、表示共享感知机Shared MLP,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点, 表示Hadamard乘积;
[0037] 使用Shared MLP将特征 、融合,获得提取后的点云特征 ,的计算公式如下:
[0038]
[0039] 其中,表示语义特征空间的特征提取、表示欧式空间的特征提取。
[0040] 优选地,在步骤S4中,对2D图像分割模块FastSCNN进行训练,过程如下:
[0041] 将训练集中的图像数据输入2D图像分割模块FastSCNN,输出图像语义分割预测值,采用交叉熵损失函数计算预测值与样本图片特征真值 之间的损失 ,计算公式如下:
[0042]
[0043] 其中,B为输入2D图像分割模块FastSCNN的图像的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别, 表示对数函数;
[0044] 对3D点云数据分割模块进行训练,过程如下:
[0045] 将训练集中的图像数据输入3D点云数据分割模块,输出点云语义分割预测值 ,采用交叉熵损失函数计算预测值与点云特征真值 之间的损失 ,计算公式如下:
[0046]
[0047] 其中,B为输入3D点云数据分割模块的数据的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别, 表示对数函数;
[0048] 训练3D点云数据分割模块,令3D点云数据分割模块输出的预测结果分布学习2D图像分割模块FastSCNN的预测结果分布,具体过程如下:
[0049] 利用知识蒸馏的方法,采用KL散度计算2D图像分割模块FastSCNN的预测结果分布U与3D点云数据分割模块的预测结果分布V的损失 ,计算公式如下:
[0050]
[0051] 其中,表示分布中具体某一类别,U表示2D图像分割模块FastSCNN的预测结果分布,V表示3D点云数据分割模块的预测结果分布, 表示对数函数;
[0052] 构建融合图像与点云数据的语义分割模型的总损失函数 , 的表达式如下:
[0053]
[0054] 最后,对总损失函数 进行优化,使损失值达到最小。
[0055] 与现有技术相比,本发明技术方案的有益效果是:
[0056] 本发明提供一种融合遥感图像与点云数据的地理要素识别方法,首先采集图像数据和点云数据并进行类别标注,构建为样本数据集,再构建融合图像与点云数据的语义分割模型,提取图像数据的2D特征和点云数据的3D特征,将2D特征和3D特征融合得到图像分割预测值和点云分割预测值,利用样本数据集训练融合图像与点云数据的语义分割模型,最后将待识别遥感图像、待识别遥感图像对应的点云数据输入训练好的融合图像与点云数据的语义分割模型,输出图像分割结果和点云分割结果,利用不同的色彩对不同类别的点云分割结果进行标记,实现地理要素的识别,充分利用了2D图像信息的有序性以及3D点云数据的空间性,提高了地理要素的分割识别精度。

附图说明

[0057] 图1表示本发明实施例1中提供的融合遥感图像与点云数据的地理要素识别方法流程示意图;
[0058] 图2表示本发明实施例1中提供的融合图像与点云数据的语义分割模型总结构图;
[0059] 图3表示本发明实施例1中提供的2D图像分割模块Fast SCNN结构图;
[0060] 图4表示本发明实施例1中提供的3D点云数据分割模块PFTransformer结构图;
[0061] 图5表示本发明实施例1中提供的3D点云数据分割模块的位置与特征自注意力块PFTransformerBlock;
[0062] 图6表示本发明实施例2中提供的3D点云数据分割模块的下采样结构Transition Down和上采样结构Transition Up示意图。

具体实施方式

[0063] 附图仅用于示例性说明,不能理解为对本专利的限制;
[0064] 为了更好地说明本实施例,附图某些部位会有省略、放大或缩小,并不代表实际尺寸;
[0065] 对于本领域技术人员来说,附图中某些公知内容说明可能省略是可以理解的。
[0066] 下面结合附图和实施例对本发明的技术方案做进一步的说明。
[0067] 附图中描述位置关系的仅用于示例性说明,不能理解为对本专利的限制。
[0068] 实施例1 如图1所示,本实施例提供一种融合遥感图像与点云数据的地理要素识别方法,所述方法包括以下步骤:
[0069] S1.采集图像数据和点云数据,对图像数据和点云数据分别进行类别标注,分别作为图像数据和点云数据的真实标签图,将带有类别标注的图像数据和点云数据构建为样本数据集,将样本数据集划分为训练集、验证集、测试集;
[0070] 在本实施例中,步骤S1还包括:对类别标注后的图像数据和点云数据进行预处理,对图像数据和点云数据进行增强处理,并在增强过程中,更新标注位置;通过像素映射对齐图像数据和点云数据;
[0071] 如图2所示,定义输入图像数据为I(Image),点云数据为P(point cloud);定义真实标签图中图像数据的样本图片特征真值为 ,点云数据的点云特征真值为 。
[0072] 在实际操作时,对图像数据的预处理的包括:对标注好的图像数据进行几何变换,包括翻转、旋转、裁剪、变形、缩放;对标注好的图像数据进行几何变换进行颜色变换,包括噪声、模糊、擦除、填充;再应用Mosaic数据增强方法,将多张图片进行随机裁剪、缩放后,随机排列拼接形成一张图片,丰富图像特征信息,并在增强过程中更新标注位置;
[0073] 对点云数据的预处理包括:基于点云空间位置的随机缩放、扰动、平移、镜像、旋转,基于色彩的对比度调整、色彩平移、色彩随机抖动、色彩随机丢失,获得经过预处理增强后的点云数据。
[0074] S2.构建融合图像与点云数据的语义分割模型,如图2所示,融合图像与点云数据的语义分割模型包括用于提取图像数据的2D特征的2D分支2D Branch(如图2所示的虚线分支部分)、用于提取点云数据的3D特征的3D分支3D Branch(如图2所示的实线单线分支部分)及融合结构Feature Fusion(如图2所示的空心箭头指示部分),2D分支2D Branch的输出端及3D分支3D Branch的输出端均连接融合结构Feature Fusion,融合结构Feature Fusion将2D特征和3D特征融合,利用2D特征优化3D特征,得到图像分割预测值和点云分割预测值;
[0075] 参见图3,所述的步骤S2中的2D分支2D Branch由2D图像分割模块Fast SCNN及2D分类器2D Classfier,其中2D图像分割模块 FastSCNN包括依次连接的主干网络特征编码器、双分支全局特征提取器、特征融合器;
[0076] 如图3所示,所述主干网络特征编码器由两个DSconv卷积块组成,进行下采样操作;双分支全局特征提取器包括两个分支:第一个分支由多个Bottleneck卷积块、SE注意力模块、Pyramid Pooling池化组成,提取图像数据的全局特征,第二个分支保留图像数据的原始特征信息,两个分支分别与特征融合器连接;特征融合器由Unsample上采样、DWConv卷积块、Conv2D卷积块组成,将全局特征提取器的两个分支的输出特征相加融合并调整通道数量,输出2D特征 。
[0077] 如图4所示,所述的3D分支3D Branch由3D点云数据分割模块PFTransformer 和3D分类器3D Classfier组成,其中3D点云数据分割模块PFTransformer包括3D点云特征编码器Encoding和3D点云特征解码器Decoding;在本实施例中,3D点云分割模块输入6维的点云3D
数据P,输出32维度的3D特征E ;
[0078] 所述3D点云特征编码器Encoding包括依次连接的五个编码块Encoder Block,分别为非下采样特征编码块和四层下采样特征编码块,参见图4,非下采样特征编码块包括共享感知机Shared MLP和特征自注意力块PFTransformer Block;每一个下采样特征编码块包括下采样结构Transition Down和特征自注意力块PFTransformer Block;
[0079] 点云数据进入非下采样特征编码块,由共享感知机Shared MLP提取共享特征,共享特征经过特征自注意力块PFTransformer Block进行自注意力增强,然后依次输入至第一层下采样特征编码块 第四层下采样特征解码块中,由各层的下采样结构Transition ~Down得到采样特征,再由特征自注意力块PFTransformer Block对采样特征进行进一步特征强化,依次完成3D点云特征编码器Encoding中第一层下采样特征编码块 第四层下采样~
特征解码块的特征编码,输出编码后的点云特征;
[0080] 所述3D点云特征解码器Decoding包括依次连接的五个解码块Decoder Block,分别为非上采样特征解码块和四层上采样特征解码块,参见图4,非上采样特征解码块包括共享感知机Shared MLP和特征自注意力块PFTransformer Block,所述上采样特征解码块包括上采样结构Transition UP和特征自注意力块PFTransformer Block;
[0081] 3D点云特征编码器Encoding输出的点云特征进入3D点云特征解码器Decoding,输入至非上采样特征解码块,首先由共享感知机Shared MLP进行特征解码,再经过特征自注意力块PFTransformer Block进行自注意力增强,增强后的点云特征依次输入至第一层上采样特征解码块 第四层上采样特征解码块中,经上采样结构Transition UP进行上采样,~并融合,获得融合后的特征,然后该特征进入特征自注意力块PFTransformer Block中,由特征自注意力块PFTransformer Block进行进一步特征强化;依次完成3D点云特征解码器中Decoding中第一层上采样特征解码块 第四层上采样特征解码块的特征解码,得到3D特~
征 ;
[0082] 所述特征自注意力块PFTransformer Block由n组循环结构构成,如图5所示,每组循环结构包含上层Shared MLP、位置与特征自注意力层PFTransformerLayer、下层Shared MLP和位置相加层Add,在本实施例中,第一层下采样特征编码块 第四层下采样特征解码块~中的特征自注意力块PFTransformer Block的循环结构数量n分别为[2,3,3,6,4],第一层上采样特征解码块 第四层上采样特征解码块中的特征自注意力块PFTransformer Block~
的循环结构的数量n分别为[2,3,3,6,4];将特征自注意力块PFTransformer Block的输入特征设为 ,坐标为 ,经过循环结构获得输出特征

[0083] 如图6所示,所述下采样结构Transition Down基于点云特征的空间位置进行最远点采样(如图6所示的Farthest point sample部分),然后通过K近邻算法对采样点进行分组(如图6所示的KNN部分),利用共享感知机Shared MLP对每个点的组内特征进行特征提取,通过局部最大池化(如图6所示的Max Pooling部分)提取组内最大特征点,得到下采样特征;
[0084] 如图6所示,所述上采样结构Transition UP利用共享感知机Shared MLP对输入的点云特征维度进行降维,然后根据点云特征的空间位置,使用三线性插值Trilinear interpolation的方法对点云特征进行上采样,得到上采样特征。
[0085] 所述位置与特征自注意力层PFTransformerLayer用于提取点云特征,过程如下:
[0086] 基于欧式空间位置的自注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ,基于欧式空间距离对N个点云特征进行提取并分别记作 ,其中,N是点云的数量,D是点云的特征维度;使用共享感知机Shared MLP对位置特征进行编码后获得特征 ;最后计算特征 , 的计算表达公式如下:
[0087]
[0088] 其中, 、 表示共享感知机Shared MLP,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点, 表示Hadamard乘积;
[0089] 基于特征空间的子注意力机制,利用3个共享感知机Shared MLP对N个点云特征进行提取并分别记作 ,基于欧式空间距离对N个点云特征进行提取并分别记作;使用Shared MLP对提取的点云特征进行编码后获得特征;最后计算特征 ,的计算表达公式如下:
[0090]
[0091] 其中,、表示共享感知机Shared MLP,Xi表示N个点中以第i个点xi为中心点构成的点云邻域集合,xj表示集合中的第j个点, 表示Hadamard乘积;
[0092] 使用Shared MLP将特征 、融合,获得提取后的点云特征 ,计算公式如下:
[0093]
[0094] 其中,表示语义特征空间的特征提取、表示欧式空间的特征提取。
[0095] 利用融合结构Feature Fusion将2D特征和3D特征融合,如图3所示,过程如下:
[0096] 3D点云数据分割模块PFTransformer输出的3D特征 经过第一共享感知机Shared MLP1后,输出为3D共享特征 ,令 与2D图像分割模块FastSCNN输出的2D特征通过Concat函数进行拼接,将拼接后的特征通过第二共享感知机Shared MLP2进行特征融合,得到融合特征 ;
[0097] 将 通过第三共享感知机Shared MLP3后再进行Sigmoid函数计算得到注意力权重,注意力权重与融合特征 进行Hadamard乘积,再以残差的方式与 利用add函数按位置相加,并通过共享感知机Shared MLP组成的2D分类器2D Classfier,获得2D特征的图像语义分割预测值;
[0098] 3D点云数据分割模块PFTransformer输出的3D特征 与经过第一共享感知机Shared MLP1得到的 进行融合,再通过3D分类器3D Classfier,获得3D点云的点云语义分割预测值;
[0099] S3.利用训练集对融合图像与点云数据的语义分割模型进行训练,训练过程中分别计算图像分割预测值和点云分割预测值与真实标签图之间的损失值,根据损失值调整融合图像与点云数据的语义分割模型的参数,直到融合图像与点云数据的语义分割模型收敛,然后利用验证集对训练过程中的融合图像与点云数据的语义分割模型进行评估,得出训练好的融合图像与点云数据的语义分割模型,并利用测试集测试融合图像与点云数据的语义分割模型的预测准确度;
[0100] S4.将待识别遥感图像、待识别遥感图像对应的点云数据输入训练好的融合图像与点云数据的语义分割模型,输出图像分割结果和点云分割结果;
[0101] S5.利用不同的色彩对不同类别的点云分割结果进行标记,展示分割结果,实现地理要素的识别。
[0102] 实施例2 在本实施例中,如图6所示,所述下采样结构Transition Down通过最远点采样FPS算法(Farthest point sample)对输入的点云特征根据空间位置分别利用欧氏空间下采样和语义特征空间下采样进行最远点采样,具体过程如下;
[0103] 输入特征 ,坐标为 ,分别在 个点中寻找欧式空间中距离最远的 个点和语义特征空间中距离最远的 个点,对每个点使用 算法,在欧式空间和特征空间中分别寻找距离最近的K个点,分别得到的K个点组成集合 和集合 ,对集合 和集合 内的每个点使用两组共享感知机Shared MLP分别集合特征进行提取;然后利用Max Pooling分别提取集合 和集合 的最大特征,并将两个集合的最大特征以相加的方式进行融合,得到采样点的上采样特征
, 的计算表达式如下:
[0104]
[0105] 其中, 表示共享感知机Shared MLP, 表示特征相加, 表示最大池化;
[0106] 所述上采样结构Transition UP根据点云特征的空间位置,使用三线性插值Trilinear interpolation的方法对点云特征进行上采样,并使用三线性插值的方法计算语义特征空间距离以加强同一空间中距离较远的同类物体的特征,具体过程如下:
[0107] 如图6所示,上采样结构Transition UP的输入特征设为 ,坐标为,上采样结构Transition UP上采样前点云的数量为 ,采样后点云的数
量为 , 、 数量与同级的Transition Down的采样点云的数量一致,i表示 个采样点中的第i个采样点,以 的第i个点为中心点,在 中使用KNN算法根据欧式距离寻找距离中心点最近的3个点,记作每个点的特征为 ,其中j=1,2,3,乘以欧式距离权重 ,作为的第i个点的上采样特征 ,的计算表达式如下:
[0108]
[0109]
[0110] 其中, 、表示 个采样点的第i个点的空间坐标,、、表示 中距离中心点最近的第j个点的空间坐标,参见图4和图6,将与3D点云特征编码器Encoding中同层的特征 通过Shared MLP后得到的 与上采样特征进行按位置相加(如图6所示的Add部分),获得输出特征 。
[0111] 实施例3 在本实施例中,对2D图像分割模块FastSCNN进行训练,过程如下:
[0112] 将训练集中的图像数据输入2D图像分割模块FastSCNN,输出图像语义分割预测值,采用交叉熵损失函数计算预测值与样本图片特征真值 之间的损失 ,计算公式如下:
[0113]
[0114] 其中,B为输入2D图像分割模块FastSCNN的图像的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别, 表示对数函数;
[0115] 对3D点云数据分割模块进行训练,过程如下:
[0116] 将训练集中的图像数据输入3D点云数据分割模块,输出点云语义分割预测值 ,采用交叉熵损失函数计算预测值与点云特征真值 之间的损失 ,计算公式如下:
[0117]
[0118] 其中,B为输入3D点云数据分割模块的数据的个数,i表示B中的第i个图像,C为类别数,j表示C中的第j个类别, 表示对数函数;
[0119] 训练3D点云数据分割模块,令3D点云数据分割模块输出的预测结果分布学习2D图像分割模块FastSCNN的预测结果分布,具体过程如下:
[0120] 利用知识蒸馏(如图2所示的Knowledge Distillation部分)的方法,采用KL散度计算2D图像分割模块FastSCNN的预测结果分布U与3D点云数据分割模块的预测结果分布V的损失 ,计算公式如下:
[0121]
[0122] 其中,表示分布中具体某一类别,U表示2D图像分割模块FastSCNN的预测结果分布,V表示3D点云数据分割模块的预测结果分布, 表示对数函数;
[0123] 构建融合图像与点云数据的语义分割模型的总损失函数 , 的表达式如下:
[0124]
[0125] 最后,利用 SGD[1]优化器对总损失函数 进行优化,使损失值达到最小。
[0126] 此外,利用训练集和验证集对融合图像与点云数据的语义分割模型进行训练后,利用测试集测试融合图像与点云数据的语义分割模型的预测准确度,具体过程如下:
[0127] 将测试集的点云数据输入融合图像与点云数据的语义分割模型,仅用3D分支3D Branch完成模型的推理过程,得到3D点云的点云语义分割预测值 ,计算 与点云特征真值 的交并比 ,计算公式如下:
[0128]
[0129] 其中,表示进行点云语义分割类别数量, 表示预测为类正确的点, 表示标签为类的点;
[0130] 将 作为融合图像与点云数据的语义分割模型的预测准确度,直到所有测试集测试完成,得到点云分割与预测的预测准确度。
[0131] 显然,本发明的上述实施例仅是为清楚地说明本发明所作的举例,而并非是对本发明的实施方式的限定。对于所属领域的普通技术人员来说,在上述说明的基础上还可以做出其它不同形式的变化或变动。这里无需也无法对所有的实施方式予以穷举。凡在本发明的精神和原则之内所作的任何修改、等同替换和改进等,均应包含在本发明权利要求的保护范围之内。