一种基于图卷积神经网络的遥感影像小水体提取方法

摘要

本发明公开了一种基于图卷积神经网络的遥感影像小水体提取方法，该方法基于我国高分2号(GF‑2)卫星影像，结合了图卷积神经网络(GCN)结构的优势，建立了一种新型的水体提取模型。包括以下步骤：对大量高分遥感影像进行水体标注，以建立训练数据集；在高分遥感影像训练集上训练建立的GCN模型；将训练好的GCN模型应用于实验区域，从模型输出的二值图像中得到水体目标。本发明能够更有效的发现面积较小的水体，极大提升了小水体的召回率，实现高精度、自动化的快速提取高分遥感影像中的水体信息。