一种基于深度学的遥感影像自动标注方法

摘要

本发明公开了一种基于深度学的遥感影像自动标注方法，该遥感影像自动标注方法首先提取待标注遥感影像的视觉特征向量，然后以视觉特征向量为DBM模型的输入对待标注遥感影像进行自动标注，该方法中实用的DBM模型由下自上依次包括可见层、第一隐层、第二隐层和标签层，该模型通过训练得到。本发明的遥感影像自动标注方法中所使用的深度玻尔兹曼机模型中包括两个隐层（分别为第一隐层和第二隐层），通过设置两个隐层有效填补图像语义标注过程中的“语义鸿沟”问题，提高整体的标注准确率。

主权项

一种基于深度学习的遥感影像自动标注方法，其特征在于，包括：（1）提取待标注遥感影像的底层特征向量构建得到相应遥感影像的视觉特征向量；（2）将所述的视觉特征向量输入训练好的深度玻尔兹曼机模型进行自动标注；所述步骤（2）中训练好的深度玻尔兹曼机模型通过以下步骤得到：（S1）创建包含若干个文本标签的标签词典；（S2）根据标签词典选择相应类别的已标注有文本标签遥感影像作为模型训练数据集；（S3）提取各张遥感影像的底层特征向量构建得到相应遥感影像的视觉特征向量，并根据标签词典和文本标签确定各个遥感影像的文本特征向量；（S4）构建深度玻尔兹曼机模型，所述的深度玻尔兹曼机模型从下至上依次包括可见层、第一隐层、第二隐层和标签层，各层中的任意两个节点无连接，相邻层之间的任意两个节点之双向连接；（S5）利用模型训练数据集中所有遥感影像的视觉特征向量和文本特征向量对所述的深度玻尔兹曼机模型进行训练，得到训练好的深度玻尔兹曼机模型。