[发明专利]一种基于随机森林的遥感影像河湖水边线提取方法

权利要求书

1.一种基于随机森林的遥感影像河湖水边线提取方法，其特征在于，包括以下步骤：

步骤1、获得遥感影像，利用随机森林对所述遥感影像进行水陆识别，获得水陆二值化图像；

步骤2、对所述水陆二值化图像进行非目标去除处理，获得目标图像；

步骤3、对所述目标图像进行河湖水边线提取，获得河湖水边线提取结果。

2.根据权利要求1所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述步骤1包括以下子步骤：

步骤1.1、根据所述遥感影像，获得所述遥感影像对应的NDWI水体指数；

步骤1.2、以所述遥感影像的红色波段、蓝色波段、NDWI水体指数作为随机森林的输入特征向量；

步骤1.3、基于所述遥感影像选取训练样本、测试样本，并对随机森林进行训练；

步骤1.4、利用训练好的随机森林对所述遥感影像进行水陆识别，获得所述水陆二值化图像。

3.根据权利要求2所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述随机森林的分类目标为水体和非水体两类，所述随机森林包含10个分类树。

4.根据权利要求2所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述随机森林采用基尼系数作为随机森林中分类树的分裂规则。

5.根据权利要求1所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述步骤2中，所述非目标包括第一类非目标、第二类非目标；所述第一类非目标为非目标水体，包括但不限于池塘、水田；所述第二类非目标为非目标船舶；

针对所述第一类非目标，采用区域生长算法进行目标水体提取，根据目标水体提取结果排除非目标水体；

针对所述第二类非目标，设置船舶大小阈值N1、内河洲滩陆地大小阈值N2，遍历所述水陆二值化图像中的像素，根据所述船舶大小阈值N1获取疑似船舶像素并作为种子点，使用区域生长算法获得疑似船舶对象集，若所述疑似船舶对象集中的像素数量小于所述内河洲滩陆地大小阈值N2，则判定疑似船舶像素为船舶对象，予以去除。

6.根据权利要求1所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述步骤3的具体实现方式为：依据水边线像素为陆地像素，且水边线像素的空间八连通区域内必然存在水体像素这一原则，对所述目标图像进行河湖水边线提取，获得河湖水边线提取结果。

7.根据权利要求1所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，还包括：步骤4、对所述河湖水边线提取结果进行精度评价。

8.根据权利要求7所述的基于随机森林的遥感影像河湖水边线提取方法，其特征在于，所述步骤4的具体实现方式为：

基于所述河湖水边线提取结果，在水边线八连通范围内随机生成M个样本点；

目视解译所述M个样本点，分为水边线点、水体点、陆地点三类，获得目视解译结果；

建立混淆矩阵，以所述目视解译结果作为参考值，以所述河湖水边线提取结果作为预测值，进行基于混淆矩阵精度评价指标的河湖水边线提取结果精度评价，获得河湖水边线提取精度评价结果。