基于堆栈自动编码机的合成孔径雷达图像分类方法
【专利摘要】基于堆栈自动编码机的合成孔径雷达图像分类方法,属于雷达图像分类【技术领域】。本发明是为了解决SAR图像数据由于噪声污染严重以及非线性特性,造成分类精度低的问题。它首先采集N幅原始SAR图像进行数据准备,获得SAR图像数据及SAR监督数据;由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据;使用训练图像数据对各层自动编码机进行预训练,在预训练的基础上,使用训练监督数据对网络进行有监督学习,微调网络,获得训练好的基于堆栈自动编码机的网络;将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类。本发明用于合成孔径雷达图像的分类。
【专利说明】基于堆栈自动编码机的合成孔径雷达图像分类方法
【技术领域】
[0001]本发明涉及基于堆栈自动编码机的合成孔径雷达图像分类方法,属于雷达图像分类【技术领域】。
【背景技术】
[0002]合成孔径雷达SAR是一种利用微波进行感知的主动传感器,相对于其它传感器,SAR可以全天候、全天时进行信息的获取。随着SAR技术的不断发展,能够获取的SAR图像越来越多,如何有效地使用这些图像成为一个亟需解决的问题。
[0003]SAR图像数据由于噪声污染严重以及其非线性特性,对其分类精度造成了限制。
【发明内容】
[0004]本发明目的是为了解决SAR图像数据由于噪声污染严重以及非线性特性,造成分类精度低的问题,提供了一种基于堆栈自动编码机的合成孔径雷达图像分类方法。
[0005]本发明所述基于堆栈自动编码机的合成孔径雷达图像分类方法,它包括以下步骤:
[0006]步骤一:采集N幅原始SAR图像进行数据准备,获得SAR图像数据及SAR监督数据;
[0007]步骤二:由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据;
[0008]步骤三:使用训练图像数据对基于堆栈自动编码机构成的网络的各层自动编码机进行预训练,学习数据特征并提取特征;在预训练的基础上,使用训练监督数据对基于堆栈自动编码机构成的网络进行有监督学习,微调网络,获得训练好的基于堆栈自动编码机的网络;
[0009]步骤四:将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类。
[0010]步骤一中获得SAR图像数据及SAR监督数据的具体方法为:
[0011]设定每幅原始SAR图像的大小为MXM,读取每一幅原始SAR图像,将每一幅原始SAR图像转化为M2的行向量;对N幅原始SAR图像进行遍历,得到SAR图像数据,为NXM2的矩阵;由原始SAR图像确定相应的图像类别标号,作为SAR监督数据,为NXl的向量,该向量中第i个元素标记第i幅原始SAR图像的类别标号,i = 1,2,3,......N。
[0012]步骤二中由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据的具体方法为:
[0013]对SAR图像数据,从第一行开始,由上至下隔行抽取行向量形成训练图像数据;从第二行开始,由上至下隔行抽取行向量形成测试图像数据;
[0014]对SAR监督数据,从第一行开始,由上至下隔行抽取行向量形成训练监督数据;从第二行开始,由上至下隔行抽取行向量形成测试监督数据。
[0015]步骤三中获得训练好的基于堆栈自动编码机的网络的具体方法为:
[0016]将训练图像数据作为基于堆栈自动编码机构成的网络的m层自动编码机的输入,m为正整数,通过非监督的学习预训练所述网络;每层自动编码机采用相同的隐层节点数目h,h为正整数,作为网络输入层的自动编码机的节点数目为M2,其余层的自动编码机的输入为上一层的输出,可视节点数目为h ;在预训练所述网络的过程中,通过不断的迭代调整网络输入层与隐层之间的连接权重,实现数据特征的学习并提取特征;
[0017]在m层自动编码机的输出端,使用逻辑回归分类器作为分类器,将作为网络输出层的自动编码机的包含有h个节点的输出作为逻辑回归分类器的输入,逻辑回归分类器的输出节点数目为原始SAR图像的类别总数C,逻辑回归分类器输出节点的相应图像类别位置置1,其它输出节点为O ;再采用训练监督数据对网络输入层与隐层之间的连接权重进行微调,获得训练好的基于堆栈自动编码机的网络。
[0018]步骤四中将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类的具体方法为:
[0019]将测试图像数据构成的二维矩阵作为训练好的基于堆栈自动编码机的网络的输入,进行分类,训练好的基于堆栈自动编码机的网络的输出为测试图像数据的图像类别标号。
[0020]将测试监督数据的类别标号与步骤四中获得的测试图像数据的图像类别标号进行比较,获得最终分类结果。
[0021]本发明的优点:本发明采用深度学习技术中的堆栈自动编码机SAE实现对合成孔径雷达图像特征的提取,它针对SAR图像数据的噪声污染严重以及非线性特性,用深度学习的方法对图像进行深层的特征学习与提取,再通过有监督的微调,结束整个深层网络的训练过程,最后将待分类的SAR图像输入网络完成分类。
[0022]本发明方法解决了 SAR图像数据的噪声污染严重以及非线性特性对分类精度的限制,充分利用了 SAR图像所包含的信息,通过深层的神经网络结构,实现了图像数据的分类。
【专利附图】
【附图说明】
[0023]图1是本发明所述基于堆栈自动编码机的合成孔径雷达图像分类方法的流程图。
【具体实施方式】
[0024]【具体实施方式】一:下面结合图1说明本实施方式,本实施方式所述基于堆栈自动编码机的合成孔径雷达图像分类方法,它包括以下步骤:
[0025]步骤一:采集N幅原始SAR图像进行数据准备,获得SAR图像数据及SAR监督数据;
[0026]步骤二:由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据;
[0027]步骤三:使用训练图像数据对基于堆栈自动编码机构成的网络的各层自动编码机进行预训练,学习数据特征并提取特征;在预训练的基础上,使用训练监督数据对基于堆栈自动编码机构成的网络进行有监督学习,微调网络,获得训练好的基于堆栈自动编码机的网络;
[0028]步骤四:将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类。
[0029]步骤一中进行数据准备是为了对原始SAR图像进行整理,以适应后续算法的需要。
[0030]【具体实施方式】二:本实施方式对实施方式一作进一步说明,步骤一中获得SAR图像数据及SAR监督数据的具体方法为:
[0031]设定每幅原始SAR图像的大小为MXM,读取每一幅原始SAR图像,将每一幅原始SAR图像转化为M2的行向量;对N幅原始SAR图像进行遍历,得到SAR图像数据,为NXM2的矩阵;由原始SAR图像确定相应的图像类别标号,作为SAR监督数据,为NXl的向量,该向量中第i个元素标记第i幅原始SAR图像的类别标号,i = 1,2,3,......N。
[0032]【具体实施方式】三:本实施方式对实施方式二作进一步说明,步骤二中由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据的具体方法为:
[0033]对SAR图像数据,从第一行开始,由上至下隔行抽取行向量形成训练图像数据;从第二行开始,由上至下隔行抽取行向量形成测试图像数据;
[0034]对SAR监督数据,从第一行开始,由上至下隔行抽取行向量形成训练监督数据;从第二行开始,由上至下隔行抽取行向量形成测试监督数据。
[0035]【具体实施方式】四:本实施方式对实施方式三作进一步说明,步骤三中获得训练好的基于堆栈自动编码机的网络的具体方法为:
[0036]将训练图像数据作为基于堆栈自动编码机构成的网络的m层自动编码机的输入,m为正整数,通过非监督的学习预训练所述网络;每层自动编码机采用相同的隐层节点数目h,h为正整数,作为网络输入层的自动编码机的节点数目为M2,其余层的自动编码机的输入为上一层的输出,可视节点数目为h ;在预训练所述网络的过程中,通过不断的迭代调整网络输入层与隐层之间的连接权重,实现数据特征的学习并提取特征;
[0037]在m层自动编码机的输出端,使用逻辑回归分类器作为分类器,将作为网络输出层的自动编码机的包含有h个节点的输出作为逻辑回归分类器的输入,逻辑回归分类器的输出节点数目为原始SAR图像的类别总数C,逻辑回归分类器输出节点的相应图像类别位置置1,其它输出节点为O ;再采用训练监督数据对网络输入层与隐层之间的连接权重进行微调,获得训练好的基于堆栈自动编码机的网络。
[0038]本实施方式包括两部分:使用训练图像数据对构成深层网络的各层自动编码机AE进行预训练,学习数据特征并提取特征;在预训练的基础上,在训练监督数据的协助下,利用反向转播算法对多层AE构成的基于堆栈自动编码机SAE进行有监督学习,微调网络。
[0039]通过不断的迭代调整网络输入层与隐层之间的连接权重,使得整个网络能够较为准确的学习并提取出数据的本质特征。
[0040]在训练监督数据的协助下,对整个网络的权重进行微调,能够保证图像数据较高的分类精度。
[0041]本实施方式中,m的取值一般为2-5 ;h的取值一般为20-100。
[0042]【具体实施方式】五:本实施方式对实施方式四作进一步说明,步骤四中将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类的具体方法为:
[0043]将测试图像数据构成的二维矩阵作为训练好的基于堆栈自动编码机的网络的输入,进行分类,训练好的基于堆栈自动编码机的网络的输出为测试图像数据的图像类别标号。
[0044]本实施方式实现了 SAR图像的分类。
[0045]【具体实施方式】六:本实施方式对实施方式五作进一步说明,将测试监督数据的类别标号与步骤四中获得的测试图像数据的图像类别标号进行比较,获得最终分类结果。
[0046]本实施方式可以计算出整个分类系统的分类精度的定量衡量。
【权利要求】
1.一种基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,它包括以下步骤: 步骤一:采集N幅原始SAR图像进行数据准备,获得SAR图像数据及SAR监督数据; 步骤二:由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据; 步骤三:使用训练图像数据对基于堆栈自动编码机构成的网络的各层自动编码机进行预训练,学习数据特征并提取特征;在预训练的基础上,使用训练监督数据对基于堆栈自动编码机构成的网络进行有监督学习,微调网络,获得训练好的基于堆栈自动编码机的网络; 步骤四:将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类。
2.根据权利要求1所述的基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,步骤一中获得SAR图像数据及SAR监督数据的具体方法为: 设定每幅原始SAR图像的大小为MXM,读取每一幅原始SAR图像,将每一幅原始SAR图像转化为M2的行向量;对N幅原始SAR图像进行遍历,得到SAR图像数据,为NXM2的矩阵;由原始SAR图像确定相应的图像类别标号,作为SAR监督数据,为NX I的向量,该向量中第i个元素标记第i幅原始SAR图像的类别标号,i = 1,2,3,......N。
3.根据权利要求2所述的基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,步骤二中由SAR图像数据获得训练图像数据和测试图像数据;由SAR监督数据获得训练监督数据及测试监督数据的具体方法为: 对SAR图像数据,从第一行开始,由上至下隔行抽取行向量形成训练图像数据;从第二行开始,由上至下隔行抽取行向量形成测试图像数据; 对SAR监督数据,从第一行开始,由上至下隔行抽取行向量形成训练监督数据;从第二行开始,由上至下隔行抽取行向量形成测试监督数据。
4.根据权利要求3所述的基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,步骤三中获得训练好的基于堆栈自动编码机的网络的具体方法为: 将训练图像数据作为基于堆栈自动编码机构成的网络的m层自动编码机的输入,m为正整数,通过非监督的学习预训练所述网络;每层自动编码机采用相同的隐层节点数目h,h为正整数,作为网络输入层的自动编码机的节点数目为M2,其余层的自动编码机的输入为上一层的输出,可视节点数目为h ;在预训练所述网络的过程中,通过不断的迭代调整网络输入层与隐层之间的连接权重,实现数据特征的学习并提取特征; 在m层自动编码机的输出端,使用逻辑回归分类器作为分类器,将作为网络输出层的自动编码机的包含有h个节点的输出作为逻辑回归分类器的输入,逻辑回归分类器的输出节点数目为原始SAR图像的类别总数C,逻辑回归分类器输出节点的相应图像类别位置置1,其它输出节点为O ;再采用训练监督数据对网络输入层与隐层之间的连接权重进行微调,获得训练好的基于堆栈自动编码机的网络。
5.根据权利要求4所述的基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,步骤四中将测试图像数据输入训练好的基于堆栈自动编码机的网络实现SAR图像分类的具体方法为: 将测试图像数据构成的二维矩阵作为训练好的基于堆栈自动编码机的网络的输入,进行分类,训练好的基于堆栈自动编码机的网络的输出为测试图像数据的图像类别标号。
6.根据权利要求5所述的基于堆栈自动编码机的合成孔径雷达图像分类方法,其特征在于,将测试监督数据的类别标号与步骤四中获得的测试图像数据的图像类别标号进行比较,获得最终分类结果。
【文档编号】G06K9/62GK104392244SQ201410765174
【公开日】2015年3月4日 申请日期:2014年12月11日 优先权日:2014年12月11日
【发明者】刘柏森, 张晔, 陈雨时 申请人:哈尔滨工业大学