基于2D-PCA和卷积神经网络的SAR图像分类方法技术

本发明专利技术公开了一种基于二维主成分分析和卷积神经网络的合成孔径雷达图像分类方法，主要解决现有技术中分类准确率低、鲁棒性差、复杂度高和噪声敏感性高的问题。其实现方案是：1.选取训练和测试样本，并进行扩增；2.对于训练样本集和测试样本集中的每个样本进行二维主成分分析，得到降维后的训练样本集和测试样本集；3.构建卷积神经网络模型，并用降维后的训练样本进行训练；4.将训练好的模型权重值加载到构建的卷积神经网络中，并将降维后的测试样本输入到该卷积神经网络中完成对原始SAR图像的分类。本发明专利技术降低模型训练复杂度和对噪声的敏感性，提高模型的鲁棒性和分类精度，可用于目标检测。

【技术实现步骤摘要】
基于2D-PCA和卷积神经网络的SAR图像分类方法
本专利技术属于图像处理领域，更进一步涉及SAR图像分类
，可用于目标检测。
技术介绍
合成孔径雷达是一种全天时，全天候工作的雷达系统，广泛应用军事，农业，导航，地理监视等诸多领域。随着雷达技术的发展，SAR图像分类已经成为了一个重要的领域。SAR特别适用于目标分类，检测以及监视等。由于SAR图像的成像机制和传统图像成像机制的差异，所以在的理解与解译的过程中SAR图像和传统图像也有很大的不同。通过人眼来分类是非常耗时耗力的，所以需要通过研究一些算法达到分类效果。随着深度学习的发展，深度学习被广泛应用在各个方面。西安电子科技大学在其申请的专利“基于深度置信网络的SAR图像目标识别分类方法”(专利号：201610136519.1，授权公告号：CN105809198A)中提出了一种合成孔径雷达图像目标识别的分类方法。该方法首先进行获取图像样本，滤波，然后提取特征，再然后训练深度置信网络对样本进行分类。该方法存在的不足之处是训练复杂度高，模型的鲁棒性差，识别率不高。SiZheChen,HaipengWang,FengXu,Ya-QiuJin，在其发表的论文“TargetClassificationUsingtheDeepConvolutionalNetworksforSARImage”中提出了一种基于深度卷积网络的SAR图像分类方法。该方法首先裁剪待分类的图像到固定的大小，然后构建5层卷积网络模型，最后利用训练集训练构建好的模型并实现测试集的分类。该方法存在的不足之处是对噪声的敏感性比较大，当随机添加部分噪声是会导致分类精度降低。
技术实现思路
本专利技术提出了一种基于二维PCA和卷积神经网络模型的合成孔径雷达图像的分类方法，可以克服上述已有技术的不足，降低模型训练复杂度和对噪声的敏感性，提高模型的鲁棒性和分类精度。为实现上述目的，本专利技术的实现方案如下：(1)从包含BMP2装甲车,BTR70装甲车,T72主战坦克数据的MSTAR数据集中选取训练和测试数据，并进行扩增：选取俯角为17度的所有图像作为训练图像X1，将每一张训练图像在水平和竖直方向随机平移(x,y)个像素点，其中x,y均为(-10,10)内的整数，重复该操作50次，使每一张训练图像扩充为50张，得到训练样本集A1；选取俯角为15度的所有图像作为测试图像X2，同样将每一张测试图像同样在水平和竖直方向随机平移(x,y)个像素点，重复该操作6次，使每一张测试图像扩充为6张，得到测试样本集A2；(2)对于训练样本集和测试样本集中的每个样本进行二维主成分分析2D-PCA，得到降维后28×28的二维矩阵：(2a)对于扩充后的训练样本和测试样本集，分别裁剪成其中心为60×60的区域；(2b)利用裁剪后的训练样本计算列方向上的投影矩阵U，根据投影矩阵U计算列方向上的降维矩阵F，再根据降维矩阵F计算行方向上的投影矩阵V；(2c)分别计算训练样本的降维矩阵Y1和测试样本的降维矩阵Y2，并保存：Y1＝VT·A1·U，Y2＝VT·A2·U，其中T表示转置；(3)搭建结构依次为：输入层，第一卷积层，第一池化层，第二卷积层，第二池化层，第一全连层，第二全连层的卷积神经网络模型，设置各层的参数，并对卷积神经网络模型进行训练；(4)利用训练好的模型权重值对测试样本进行分类：(4a)将训练好的模型权重值加载到卷积神经网络模型中；(4b)将测试样本Y2输入到(5a)的卷积神经网络模型中，得到所有的测试样本类别概率P，每一个样本的类别概率包含三个值，分别对应该样本是BMP2，BTR70，T72的概率；(4c)从P中取连续6个测试样本的类别概率Pi×6,...,(i+1)×6，共计有18个概率值，从这18个概率值中取其最大值对应的类别作为测试图片X2中第i个图片的最终分类结果，其中i＝1…n，n为测试图片的数量。本专利技术与现有的技术相比具有以下优点：1、本专利技术将SAR图像进行主成分分析，减少了卷积网络模型的输入数据的维度，降低了模型的复杂度和对噪声的敏感性；2、本专利技术对测试集进行扩增融合，充分利用了模型平移不变性，显著提升了模型的分类准确率；附图说明图1是本专利技术的实现流程图；图2是本专利技术中输入的部分MSTAR目标图像；图3是本专利技术中测试样本添加噪声后的图像。具体实施方式本专利技术的实现方案是：选取训练图像和测试图像并进行扩增，利用二维主成分分析对扩增后的数据降维，搭建卷积神经网络模型，用降维后的训练样本训练卷积神经网络模型，最终对测试样本进行分类。参照图1，本专利技术的具体实现步骤如下：步骤1.选取训练和测试样本，并进行扩增。1.1)从包含BMP2装甲车,BTR70装甲车,T72主战坦克数据的MSTAR数据集中选取数据集中俯角为17度的所有图像共1617张，作为训练图像X1，将每一张训练图像在水平和竖直方向随机平移(x,y)个像素点，其中x,y均为(-10,10)内的整数，重复该操作50次，使每一张训练图像扩充为50张，得到训练样本集A1；1.2)选取数据集中俯角为15度的所有图像共1365张，作为测试图像X2，同样将每一张测试图像同样在水平和竖直方向随机平移(x,y)个像素点，重复该操作6次，使每一张测试图像扩充为6张，得到测试样本集A2。步骤2.对于训练样本集和测试样本集中的每个样本进行二维主成分分析2D-PCA降维。2.1)将扩充后的训练样本和测试样本集，分别裁剪成其中心为60×60的区域；2.2)计算训练样本A1列方向的协方差矩阵G：其中，A1i为第i个训练样本，N为训练样本的数量，为全体训练样本的平均值，T表示转置；2.3)通过如下等式计算训练样本A1列方向的协方差矩阵G的特征值λ1和特征向量T1：det(G-λ1I)＝0，(G-λ1I)T1＝0，其中，I表示单位矩阵，det(G-λ1I)表示矩阵G-λ1I的行列式；2.4)从2.3)的结果中取前28个最大的特征值对应的特征向量组成列方向上投影矩阵U；2.5)根据投影矩阵U计算列方向上的降维矩阵F：F＝A1·U，其中A1为训练样本集；2.6)计算降维矩阵F行方向上的的协方差矩阵W：其中，Fi为第i个列方向上的降维矩阵，为全体列方向上的降维矩阵的平均值；2.7)通过如下等式计算降维矩阵F行方向上的协方差矩阵W的特征值λ2和特征向量T2，det(W-λ2I)＝0，(W-λ2I)T2＝0，其中，I表示单位矩阵，det(W-λ2I)表示矩阵W-λ2I的行列式；2.8)从2.7)的结果中取前28个最大的特征值对应的特征向量组成行方向上投影矩阵V；2.9)分别计算训练样本的降维矩阵Y1和测试样本的降维矩阵Y2，并保存：Y1＝VT·A1·U，Y2＝VT·A2·U。步骤3.构建卷积神经网络模型。3.1)设卷积神经网络共为7层，其结构依次为：输入层，第一卷积层，第一池化层，第二卷积层，第二池化层，第一全连层，第二全连层的卷积神经网络模型；3.2)设置各层的参数：输入层的输入尺寸为28*28*1；第一卷积层特征图的总数为32，卷积核大小为5*5，步长为1*1；第一池化层池化核大小为2*2，步长为2*2；第二卷积层特征图的总数为64，卷积核大小为5*5，步长为1*1；第二池化层池化核大小为2*2，步长为本文档来自技高网...

【技术保护点】
1.一种基于2D‑PCA和卷积神经网络的SAR图像分类方法，包括如下：(1)从包含BMP2装甲车,BTR70装甲车,T72主战坦克数据的MSTAR数据集中选取训练和测试数据，并进行扩增：选取俯角为17度的所有图像作为训练图像X1，将每一张训练图像在水平和竖直方向随机平移(x,y)个像素点，其中x,y均为(‑10,10)内的整数，重复该操作50次，使每一张训练图像扩充为50张，得到训练样本集A1；选取俯角为15度的所有图像作为测试图像X2，同样将每一张测试图像同样在水平和竖直方向随机平移(x,y)个像素点，重复该操作6次，使每一张测试图像扩充为6张，得到测试样本集A2；(2)对于训练样本集和测试样本集中的每个样本进行二维主成分分析2D‑PCA降维，得到降维后28×28的二维矩阵：(2a)对于扩充后的训练样本和测试样本集，分别裁剪成其中心为60×60的区域；(2b)利用裁剪后的训练样本计算列方向上的投影矩阵U，根据投影矩阵U计算列方向上的降维矩阵F，再根据降维矩阵F计算行方向上的投影矩阵V；(2c)分别计算训练样本的降维矩阵Y1和测试样本的降维矩阵Y2，并保存：Y1＝VT·A1·U，Y2＝VT·A2·U，其中T表示转置；(3)搭建结构依次为：输入层，第一卷积层，第一池化层，第二卷积层，第二池化层，第一全连层，第二全连层的卷积神经网络模型，设置各层的参数，并对卷积神经网络模型进行训练；(4)利用训练好的模型权重值对测试样本进行分类：(4a)将训练好的模型权重值加载到卷积神经网络模型中；(4b)将测试样本Y2输入到(5a)的卷积神经网络模型中，得到所有的测试样本类别概率P，每一个样本的类别概率包含三个值，分别对应该样本是BMP2，BTR70，T72的概率；(4c)从P中取连续6个测试样本的类别概率Pi×6,...,(i+1)×6，共计有18个概率值，从这18个概率值中取其最大值对应的类别作为测试图片X2中第i个图片的最终分类结果，其中i＝1…n，n为测试图片的数量。...

【技术特征摘要】
1.一种基于2D-PCA和卷积神经网络的SAR图像分类方法，包括如下：(1)从包含BMP2装甲车,BTR70装甲车,T72主战坦克数据的MSTAR数据集中选取训练和测试数据，并进行扩增：选取俯角为17度的所有图像作为训练图像X1，将每一张训练图像在水平和竖直方向随机平移(x,y)个像素点，其中x,y均为(-10,10)内的整数，重复该操作50次，使每一张训练图像扩充为50张，得到训练样本集A1；选取俯角为15度的所有图像作为测试图像X2，同样将每一张测试图像同样在水平和竖直方向随机平移(x,y)个像素点，重复该操作6次，使每一张测试图像扩充为6张，得到测试样本集A2；(2)对于训练样本集和测试样本集中的每个样本进行二维主成分分析2D-PCA降维，得到降维后28×28的二维矩阵：(2a)对于扩充后的训练样本和测试样本集，分别裁剪成其中心为60×60的区域；(2b)利用裁剪后的训练样本计算列方向上的投影矩阵U，根据投影矩阵U计算列方向上的降维矩阵F，再根据降维矩阵F计算行方向上的投影矩阵V；(2c)分别计算训练样本的降维矩阵Y1和测试样本的降维矩阵Y2，并保存：Y1＝VT·A1·U，Y2＝VT·A2·U，其中T表示转置；(3)搭建结构依次为：输入层，第一卷积层，第一池化层，第二卷积层，第二池化层，第一全连层，第二全连层的卷积神经网络模型，设置各层的参数，并对卷积神经网络模型进行训练；(4)利用训练好的模型权重值对测试样本进行分类：(4a)将训练好的模型权重值加载到卷积神经网络模型中；(4b)将测试样本Y2输入到(5a)的卷积神经网络模型中，得到所有的测试样本类别概率P，每一个样本的类别概率包含三个值，分别对应该样本是BMP2，BTR70，T72的概率；(4c)从P中取连续6个测试样本的类别概率Pi×6,...,(i+1)×6，共计有18个概率值，从这18个概率值中取其最大值对应的类别作为测试图片X2中第i个图片的最终分类结果，其中i＝1…n，n为测试图片的数量。2.根据权利要求书1中所述的方法，其中(3)中搭建的卷积神经网络模型中，各层的参数设置如下：输入层的输入尺寸为28*28*1；第一卷积层特征图的总数为32，卷积核大小为5*5，步长为1*1；第一池化层池化核大小为2*2，步长为2*2；第二卷积层特征图的总数为64，卷积核大小为5*5，步长为1*1；第二池化层池化核大小为2*2，...

【专利技术属性】
技术研发人员：侯彪，李井亮，焦李成，马晶晶，马文萍，白静，
申请(专利权)人：西安电子科技大学，
类型：发明
国别省市：陕西,61