[发明专利]一种基于U-net与Seg-net网络融合的航空地面建筑物提取方法

权利要求书

1.一种基于U-net与Seg-net网络融合的航空地面建筑物提取方法，包括以下步骤：

步骤S1、训练样本制作：

选择现有的图像数据库，进行裁剪及处理，获得训练样本；

步骤S2、深度学习模型训练：选取U-net和Seg-net网络的框架，使用训练样本，分别进行模型训练，最后获得两个训练模型；

步骤S3、测试训练模型：

步骤S301、首先对原图像进行切割，然后通过步骤S2得到的模型对每一张图像进行前向预测分割，获得两张MASK结果图像；

步骤S302、对两张结果图像的每个像素点采取投票表决其类别，票数最多的类别即为该像素点的类别；

步骤S4、输出结果：

输出二值化的标记分割结果，通过标记位置获取地面建筑物在原图中的位置。

2.如权利要求1所述的方法，其特征在于，所述步骤S1的训练样本中含有标注信息。

3.如权利要求1所述的方法，其特征在于，所述步骤S1的训练样本做裁剪、旋转、镜像、光照调整、增加噪声操作。

4.如权利要求1所述的方法，其特征在于，所述步骤S3还利用形态学处理孤立像素点的分类或修复断裂边缘。

5.如权利要求1所述的方法，其特征在于，所述步骤S3中对原图像的切割采用有重叠的裁剪方式。

6.如权利要求1所述的方法，其特征在于，所述步骤S302中投票为，两张结果图中相同位置像素点的类别以U-net预测的类别为优先级进行最终类别的决策。