[发明专利]结构重构方法、装置、电子设备及存储介质

权利要求书

1.一种结构重构方法，其特征在于，包括：

获取第一图像；

获取第一图像的第一特征图和第二特征图；

利用角点检测模型对第一特征图处理以获得角点特征图；

利用提议提取模型基于第一特征图、所述角点检测模型获得角点特征图的过程中产生的中间特征和角点特征图获得提议特征；

利用边检测模型基于所述第二特征图、所述角点特征图和所述提议特征获得第一线段特征图。

2.根据权利要求1所述的结构重构方法，其特征在于，所述第一图像为光栅图或遥感图。

3.根据权利要求1所述的结构重构方法，其特征在于，所述第一线段特征图包括如下线段的信息：

表征第一图像中物体结构边缘特征的线段；

属于第一图像中物体结构上同一封闭区域的区域线段；

跨域第一图像中物体结构上不同封闭区域的非法线段。

4.根据权利要求1所述的结构重构方法，其特征在于，所述第一图像中的物体包括建筑物。

5.根据权利要求1所述的结构重构方法，其特征在于，所述第二特征图的尺度低于所述第一特征图。

6.根据权利要求1所述的结构重构方法，其特征在于，所述第二特征图包括尺度不同的多张特征图；和/或，所述第一特征图包括尺度不同的多张特征图。

7.根据权利要求1所述的结构重构方法，其特征在于，所述边检测模型包括依次连接的自注意力模块、交叉注意力模块和前馈神经网络。

8.根据权利要求1～7任一项所述的结构重构方法，其特征在于，利用边检测模型基于所述第二特征图、所述角点特征图和所述提议特征获得第一线段特征图，包括：

将所述角点特征图与所述提议特征融合，以获得角点提议特征；

利用边检测模型对所述第二特征图和所述角点提议特征处理，以获得第一线段特征图。

9.根据权利要求8所述的结构重构方法，其特征在于，将所述角点特征图与所述提议特征融合，包括：

获取所述角点特征图的位置嵌入特征；

对所述位置嵌入特征和所述提议特征执行求和处理，获得角点提议特征。

10.根据权利要求8所述的结构重构方法，其特征在于，利用边检测模型基于所述第二特征图、所述角点特征图和所述提议特征获得第一线段特征图，包括：

利用所述边检测模型中的边编码器对所述第二特征图处理以获得第一编码特征；

利用所述边检测模型中的边解码器基于角点提议特征和第一编码特征获得第一线段特征图。

11.根据权利要求10所述的结构重构方法，其特征在于，利用边检测模型基于所述第二特征图、所述角点特征图和所述提议特征获得第一线段特征图，包括：

以角点提议特征作为查询向量、以角点提议特征经第一线性变换得到的特征作为键向量、以角点提议特征经第二线性变换得到的特征作为值向量，执行所述边解码器中自注意力模块的处理，获得第一注意力特征；

以第一注意力特征作为查询向量、以第一编码特征经第三线性变换得到的特征作为键向量、以第一编码特征经第四线性变换得到的特征作为值向量，执行所述边解码器中交叉注意力模块的处理，以获得第二注意力特征；

利用所述边解码器中的前馈神经网络对所述第二注意力特征处理，获得第一线段特征图。

12.根据权利要求1～6任一项所述的结构重构方法，其特征在于，利用角点检测模型对第一特征图处理以获得角点特征图，包括：

利用角点检测模型中的角点模块对所述第一特征图处理，获得第三特征图；

将第一特征图与第三特征图融合得到第四特征图；

利用角点检测模型中的角点方向模块对第四特征图处理，获得角点特征图。