一种基于深度学习的遥感图像云检测方法与流程

文档序号:15935832发布日期:2018-11-14 02:22阅读:3419来源:国知局

本发明涉及图像处理、深度学习、目标检测领域,特别是一种基于深度学习的遥感图像云检测方法。



背景技术:

近几年来,随着遥感图像获取技术的发展,遥感图像被广泛应用到各个领域,为更好解读图像信息,由于天气因素出现在地面目标上方的云成为了研究的热点。遥感图像的云检测问题可以看做是图像分析的二值分割问题。卫星云图千变万化,云的高度、厚度、种类以及太阳高度角等因素均会对云产生极大的影响。就目前来看,可以将其检测方法分为以下几类。

物理检测的方法是使用多光谱物理特性应用到单个像素上进行检测,通过综合考虑集中光谱谱段进行阈值检测,其关键及时在于物理阈值的选取。基于遥感图像中云的纹理和空间特性来进行检测也是一个有效的途径,图像中云的存在明显增加了辐射度的空间变化性,也对图像的纹理产生了很大影响,目前已经得到实际应用。但当云层的厚度较薄或者背景较复杂的时候,上述方法难以取得较好的检测结果。

随着科技的迅速发展以及大数据时代的到来,一些传统的图像识别算法已经很难适应新形势下的图像检测与识别需求,基于深度学习的人工神经网络技术的出现很好的解决了上述问题,特有的非线性映射、自学习、自组织的信息处理能力,可以深度挖掘图像的特征信心,使得在遥感图像的很多领域大放异彩,具有广阔的应用前景,已经成为一大研究热点。



技术实现要素:

本发明所要解决的技术问题是,针对现有技术不足,提供一种基于深度学习的遥感图像云检测方法,采用深度学习的方法来深度挖掘遥感图像中云的特征信息,克服原有检测技术在特征提取上不够深入的缺点,提高检测精度和算法的普适性。

为解决上述技术问题,本发明所采用的技术方案是:一种基于深度学习的遥感图像云检测方法,其特征在于,包括以下步骤:

1)对获得的遥感图像数据集进行图像预处理,并将预处理后的图像分为训练集和测试集;

2)将训练集中的图像作为神经网络模型的输入,不断更新网络模型参数优化神经网络模型;

3)将测试集图像输入到训练好的神经网络模型中进行验证,将遥感图像中有云的部分检测出来。

步骤1)的具体实现过程包括:从landsat-8卫星图像中下载9个数据集作为实验数据,将所述实验数据中的近红外、短波红外和热红外三个波段的图像进行图像融合,将融合后的图像裁剪为256*256大小的子图,然后进一步降采样到64*64大小,再讲降采样后的图像分为训练集和测试集。

步骤2)中卷积神经网络的训练算法和传统的反向传播算法差不多,主要分为4步来更新参数和优化模型,这4步可以分为2个阶段。

第一阶段:前向传播阶段。

a)从测试样本集中取一个样本(x,y),将x输入网络;x是指样本本身信息,y为样本所对应的标签值;

b)计算相应的实际输出

在此阶段,信息从输入层经过逐级的变换,传送到输出层。这个过程也是网络在完成训练后正常运行时执行的过程。在此过程中,网络执行的是计算(实际上就是输入与每层的权值矩阵相点乘,得到最后的输出结果):

其中,fn表示第n层卷积操作,wn表示第n层网络的权值矩阵;

第二阶段:后向传播阶段。

a)计算实际输出和相应的理想输出y的差值;

b)按照极小化误差的方法反向传播调整权值矩阵。

所述神经网络模型包括6层,前三层为卷积层,后三层为全连接层。

与现有技术相比,本发明所具有的有益效果为:本发明在结合使用深度学习卷积神经网络的基础上,解决了由于云特征提取不够充分而导致检测结果不够理想的问题,提高了检测精度,加强了算法的普适性。随着深度学习的不断推广与普及和遥感图像的进一步发展,本发明方法有着较大的意义与实用价值。

附图说明

图1为本发明算法的流程结构图;

图2为本发明算法的卷积神经网络模型图;conv表示卷积层,fc表示全连接层;

图3为两张测试图片;其中,(a)为有云的测试图片;(b)为无云的测试图片。

具体实施方式

本发明的方法首先对获得的遥感图像数据集进行图像预处理,划分为训练集和测试集,在用训练集对卷积神经网络进行训练的过程中,不断优化网络模型,最后用测试集图像进行验证,实现对遥感图像的高精度云检测。本发明的方法结构图如图1所示,具体方法可以分为以下步骤:

1.从landsat-8卫星图像中下载了9个数据集作为实验数据。借助envi软件平台将数据集中的近红外、短波红外和热红外三个波段的图像进行图像融合,将融合后的图像裁剪为256*256大小的子图,然后进一步降采样到64*64大小。从处理后的图像数据集中随机选取70%的图像作为训练集,剩余的30%作为测试集。训练集用于训练神经网络模型,测试集用于进行验证。

2.将训练集数据作为神经网络的输入进行模型训练,卷积神经网络的框架结构图如图2所示,一共包含6层,前三层为卷积层,后三层为全连接层。对卷积层来说,特征图的大小定义为宽度、高度和深度的乘积,前两个维度描述空间大小,深度定义通道数量。将遥感图像的云检测问题看做是一个二分类问题,因此最后输出层的数量是2。用conv表示卷积层,maxp表示池化层,fc表示全连接层,relu作为激活函数,则整个网络架构可以描述如下:

conv1(64*64*32)→relu1→maxp1→

conv2(32*32*48)→relu2→maxp2→

conv3(16*16*64)→relu3→maxp3→

fc1(128)→relu4→fc2(64)→relu5→fc3(2).

在每个分支中,卷积层的滤波器大小是3*3,池化操作是在卷积层输出的基础上,在3*3的邻域像素空间选择最大值进行,步长为2。

3.保存训练好的模型参数,将测试集数据输入到该网络中,输出为预测结果:有云或无云。将预测结果和图像真实信息进行对比,验证模型的检测效果。图3是两张测试图片,分别是有云和无云。结果表明该方法具有较高的准确度。



技术特征:

技术总结
本发明公开了一种基于深度学习的遥感图像云检测方法,其包括以下步骤:对获得的遥感图像数据集进行图像预处理,并将预处理后的图像分为训练集和测试集;将训练集中的图像作为神经网络模型的输入,不断更新网络模型参数优化神经网络模型;将测试集图像输入到训练好的神经网络模型中进行验证,将遥感图像中有云的部分检测出来。本发明在结合使用深度学习卷积神经网络的基础上,解决了由于云特征提取不够充分而导致检测结果不够理想的问题,提高了检测精度,加强了算法的普适性。

技术研发人员:杨俊刚;安成锦;曾晓双;李骏;安玮
受保护的技术使用者:中国人民解放军国防科技大学
技术研发日:2018.04.28
技术公布日:2018.11.13
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1