[发明专利]建筑物轮廓提取方法、系统、计算机设备及存储介质

权利要求书

1.一种建筑物轮廓提取方法，其特征在于：所述建筑物轮廓提取方法的步骤包括：

获取待处理影像并输入至预设的实例分割模型中，得到待处理影像中的建筑物掩膜与建筑边界框；

根据所述建筑边界框获取目标建筑的影像块，将所述影像块与对应的建筑物掩膜输入图像分割模型，得到目标建筑的建筑物图斑，获取建筑物图斑的轮廓像素点位置；

基于目标建筑的影像块检测目标建筑的主方向，基于所述主方向对目标建筑的建筑物图斑进行边界规则化处理，生成建筑物边界提取结果。

2.根据权利要求1所述的一种建筑物轮廓提取方法，其特征在于：获取待处理影像并输入至预设的实例分割模型中，得到待处理影像中的建筑物掩膜与建筑边界框的步骤之前，还包括：

获取建筑物样本图像并生成建筑物样本图像集；

通过建筑物样本图像集训练得到所述实例分割模型。

3.根据权利要求1所述的一种建筑物轮廓提取方法，其特征在于：将所述影像块与对应的建筑物掩膜输入图像分割模型，得到目标建筑的建筑物图斑的步骤中，具体包括：

基于建筑物掩膜图像在所述影像块中标记前景背景，生成前景背景标记图像；

基于前景背景标记图像获取高斯混合模型的参数；

基于高斯混合模型的参数，计算建筑物图斑的边界项，通过min-cut/max-flow算法将能量函数最小化，得到建筑物图斑分割结果。

4.根据权利要求3所述的一种建筑物轮廓提取方法，其特征在于：基于高斯混合模型的参数，计算建筑物图斑的边界项，通过min-cut/max-flow算法将能量函数最小化，得到建筑物图斑分割结果的步骤之后，还包括：

基于建筑物图斑分割结果的前景和背景重新获取高斯混合模型的参数，并重新获取建筑物图斑分割结果，重复本步骤至算法收敛。

5.根据权利要求1所述的一种建筑物轮廓提取方法，其特征在于：基于目标建筑的影像块检测目标建筑的主方向的步骤中，包括：

通过LSD算法对目标建筑的影像块进行处理，得到目标建筑的主方向。

6.根据权利要求5所述的一种建筑物轮廓提取方法，其特征在于：设置方向误差阈值e，基于所述主方向对目标建筑的建筑物图斑进行边界规则化处理，生成建筑物边界提取结果的步骤中，包括：

对获取的建筑物图斑的轮廓像素点坐标进行抽稀处理得到点集S，计算建筑物主方向的斜率k和主方向的垂直方向的斜率v；

按顺序遍历点集S所有的点，依序计算点p_i与相邻点p_i+1构成的直线的斜率d_i；

当|d_i-k|e或者|d_i-v|e时，从点p_i开始遍历点集S，以获取主方向线段正交的相邻线段，从而得到建筑物图斑的规则边界作为建筑物边界提取结果。

7.根据权利要求6所述的一种建筑物轮廓提取方法，其特征在于：基于目标建筑的影像块检测目标建筑的主方向，基于所述主方向对目标建筑的建筑物图斑进行边界规则化处理，生成建筑物边界提取结果的步骤之后，还包括：

获取待处理影像的成像信息，基于所述成像信息和所述建筑物边界提取结果计算待处理影像中建筑物的面积数据。

8.一种建筑物轮廓提取系统，其特征在于，包括：

建筑识别模块，用于获取待处理影像并输入至预设的实例分割模型中，得到待处理影像中的建筑物掩膜与建筑边界框；

图像分割模块，用于根据所述建筑边界框获取目标建筑的影像块，将所述影像块与对应的建筑物掩膜输入图像分割模型，得到目标建筑的建筑物图斑，获取建筑物图斑的轮廓像素点位置；

边界规则化处理模块，用于基于目标建筑的影像块检测目标建筑的主方向，基于所述主方向对目标建筑的建筑物图斑进行边界规则化处理，生成建筑物边界提取结果。