一种基于图卷积神经网络的遥感影像地物识别方法与流程

[0001]
本发明涉及基于遥感卫星成像技术的地物类型识别领域，具体为一种基于图卷积神经网络的遥感影像地物识别方法。


背景技术：

[0002]
随着人工智能相关技术飞速发展，深度学习在很多领域取得突破性成果。其中卷积神经网络依靠其局部特征提取、参数共享、全局池化等特点在图像识别中取得较为突出的成绩。在遥感影像处理领域，基于卷积神经网络的遥感影像分类是目前的研究热点之一。其分类精度较高，泛化能力较强。但同时也存在一些问题。在模型构建方面，用于分类的卷积神经网络模型的网络层次结构很少被直接调整，而是采用经典卷积神经网络模型对分类样本进行特征提取，再结合传统遥感影像分类模型进行分类工作，对卷积神经网络结构调整及迁移学习在遥感影像分类中的相关研究及应用较少。由于遥感影像包含信息量较多，而且数据应用情况远滞后于其数据获取能力，所以如何把这些海量的图像进行快速的分类是加快遥感图像利用率的关键所在。近来，越来越多的研究开始将深度学习方法应用到图数据领域。近年来，受到卷积神经网络在计算机视觉领域所获巨大成功的激励，涌现出很多基于谱图数据重新定义卷积概念的方法。基于图卷积神经网络聚集近邻节点的信息，直接在图结构上执行卷积。再通过采样策略，计算可以在一个批量的节点而不是整个图中执行，可以有效提高效率。然而，如何利用基于图卷积神经网络应用于遥感影像并构建高精度分类器还亟需进一步研究与发展。因此，本发明提供了一种基于图卷积神经网络的遥感影像地物分类识别，能够根据一部分的地物真实数据精确地识别出地物类型。


技术实现要素：

[0003]
本发明的目的在于依据遥感卫星成像技术提供一种基于图卷积神经网络的遥感影像地物识别方法，基于不同遥感影像训练数据集，首先提取训练数据集中地物类型特征图谱，形成各个地物类别特征描述集，进而采用深度卷积神经网络对遥感影像地物特征描述集中各个类别特征图谱进行学习与融合，获取该地物类型节点、边等特征描述，再采用图卷积神经网络对融合后的图特征进行节点分类模型训练，确定遥感影像分类模型参数，最终实现对未标记影像地物的精确识别。
[0004]
为实现上述目的，本发明提供如下技术方案：一种基于图卷积神经网络的遥感影像地物识别方法，该方法包含以下步骤：
[0005]
步骤1：取一景遥感影像分别进行相关预处理操作，包括几何校正、大气校正和图像增强，将彩色遥感影像转化为xy坐标下的灰度图像或者cielab颜色空间，对其进行简单线性迭代聚类，初始化种子点，生成紧凑、近似均匀的超像素；
[0006]
步骤2：基于生成的超像素遥感影像图，创建图谱的节点和边，对影像进行提取纹理特征、颜色特征作为图谱输入特征矩阵；
[0007]
步骤3：结合集成学习的绝大多数投票机制进行标签制作，若某地物标记得票超过
一半，则将超像素预测为该地物类别，否则拒绝标记，进而形成图谱的标签数据集；
[0008]
步骤4：利用深度卷积神经网络对高分辨率遥感影像特征图谱自动提取，得到关于影像图的节点和边输入数据，形成关于遥感影像特征描述集；
[0009]
步骤5：基于遥感影像特征描述集，训练图卷积神经网络模型，获得影像地物分类模型并对测试样本分类。
[0010]
作为本发明的一种优选技术方案，所述步骤1-4中是对遥感影像提取其地物类别特征描述集，所述步骤5中提出的是基于图卷积神经网络的地物分类模型。
[0011]
本发明的有益效果是：本发明通过首先利用超像素分割和绝大多数投票机制来构造遥感影像地物类别特征描述集，获得关于遥感影像地物类别近似刻画，同时大规模约减了同一幅影像图谱构建的数据量，有利于处理大数据样本下的遥感影像地物分类问题；最后通过融合型特征图谱集训练图卷积神经网络，从而达到精确区分地物类型的目的。本发明将非常有利于遥感自然灾害监测、土地覆盖类型判别、城市规划和生态环境变化监测等方面的研究。
附图说明
[0012]
图1为本发明的流程简图；
[0013]
图2为本发明的具体流程图。
具体实施方式
[0014]
下面结合附图对本发明的较佳实施例进行详细阐述，以使本发明的优点和特征能更易被本领域人员理解，从而对本发明的保护范围做出更为清楚明确的界定。
[0015]
实施例：请参阅图1-2，本发明提供一种技术方案：一种基于图卷积神经网络的遥感影像地物识别方法，该方法包含以下步骤：
[0016]
步骤1：取一景遥感影像分别进行相关预处理操作，包括几何校正、大气校正和图像增强，将彩色遥感影像转化为xy坐标下的灰度图像或者cielab颜色空间，对其进行简单线性迭代聚类，初始化种子点，生成紧凑、近似均匀的超像素；
[0017]
步骤2：基于生成的超像素遥感影像图，创建图谱的节点和边，对影像进行提取纹理特征、颜色特征作为图谱输入特征矩阵；
[0018]
步骤3：结合集成学习的绝大多数投票机制进行标签制作，若某地物标记得票超过一半，则将超像素预测为该地物类别，否则拒绝标记，进而形成图谱的标签数据集；
[0019]
步骤4：利用深度卷积神经网络对高分辨率遥感影像特征图谱自动提取，得到关于影像图的节点和边输入数据，形成关于遥感影像特征描述集；
[0020]
步骤5：基于遥感影像特征描述集，训练图卷积神经网络模型，获得影像地物分类模型并对测试样本分类。
[0021]
步骤1-4中是对遥感影像提取其地物类别特征描述集，步骤5中提出的是基于图卷积神经网络的地物分类模型。
[0022]
为方便描述，首先对本发明中特异名词作如下定义：
[0023]
遥感影像地物类别特征描述集：
[0024]
遥感影像地物类别特征描述集在本发明中指：基于深度卷积网络提取的图节点、
边的特征图谱，从某被超像素分割后块影像中提取出不同地物类型的图特征图谱，形成基于图谱的遥感影像地物类型特征描述集，来近似描述该影像的类别信息。
[0025]
该具体方法包含以下步骤：
[0026]
步骤1：遥感影像图特征描述集的形成过程；
[0027]
取一景遥感影像分别进行相关预处理操作，包括了几何校正，大气校正和图像增强等，假定将遥感影像分成m个块，逐块处理遥感影像，将对应的影像真实地物标签一同进行导入，将彩色图像转化为xy坐标下的灰度图像，对其进行简单线性迭代聚类，初始化种子点(聚类中心)，按照设定的超像素个数，在图像内均匀的分配种子点；然后对灰度图像构成的特征向量构造距离度量标准，对分块图像像素进行局部聚类过程，生成紧凑、近似均匀的n个超像素，接着对生成的超像素创建图谱的节点和边，对影像进行设置局部二值模式，将超像素与其邻近的超像素进行比较，并把结果保存为二进制数，形成纹理特征，作为图谱输入的特征；再结合绝大多数投票机制进行标签制作，若某地物标记得票超过一半，则将超像素预测为该地物类别，否则拒绝标记，由此可见，该遥感影像地物识别问题被建模成一个分类问题；
[0028]
步骤2：深度卷积神经网络自动提取遥感影像特征图谱集过程：假定共有k种地物类别卷积神经网络自动提取的特征图谱集形成可表示为其中其中1≤k≤m表示第k块分块遥感影像,n(1≤k≤n)表示第n个超像素,x＝cnn(
·
)表示执行深度卷积神经网络；
[0029]
步骤3：基于图卷积神经网络的遥感影像地物识别分类过程：根据全连接卷积神经网络自动提取的特征图谱，假定地物类别总共有k种，则其地物识别过程可由以下几个步骤构成；
[0030]
步骤3.1：由步骤2生成的遥感影像地物类别特征图谱集1≤k≤m，联接生成融合的特征图谱，对图卷积神经网络进行训练，并采用留一法执行交叉验证流程，获取图卷积神经网络的地物分类模型训练参数θ，其中图卷积神经网络层数为2层，并在其后加上自适应注意力机制层，使图中的每个节点可以根据相邻节点的特征，为其分配不同的权值，损失函数选用nll_loss,对输入参数进行log_softmax函数激活，优化函数选用一种自适应学习率的优化方法adam优化函数进行优化，利用梯度的一阶矩估计和二阶矩估计动态的调整学习率；
[0031]
步骤3.2：将加上了注意力机制的图卷积神经网络提取的自适应特征描述集及其注意力特征权值进行一次不含注意力机制层的图卷积操作并进行归一化处理，最后依据分类模型确定各个节点所属类别，其公式化描述为o
pre
＝gcn(
·
),其中o
pre
表示输出所属类别，gcn(
·
)表示执行图卷积神经网络操作；
[0032]
步骤3.3、将输出后的结果与真实地物类别进行精度运算，求出相应的kappa系数，总体精度以及混淆矩阵等度量参数，进而判断模型分类效果。
[0033]
本发明通过首先利用超像素分割和绝大多数投票机制来构造遥感影像地物类别特征描述集，获得关于遥感影像地物类别近似刻画，同时大规模约减了同一幅影像图谱构建的数据量，有利于处理大数据样本下的遥感影像地物分类问题；最后通过融合型特征图
谱集训练图卷积神经网络，从而达到精确区分地物类型的目的。本发明将非常有利于遥感自然灾害监测、土地覆盖类型判别、城市规划和生态环境变化监测等方面的研究。
[0034]
上实施例仅表达了本发明的几种实施方式，其描述较为具体和详细，但并不能因此而理解为对发明专利范围的限制。应当指出的是，对于本领域的普通技术人员来说，在不脱离本发明构思的前提下，还可以做出若干变形和改进，这些都属于本发明的保护范围。