基于DCCGAN模型的极化SAR图像分类方法与流程

本发明属于图像处理
技术领域：
，更进一步涉及极化合成孔径雷达图像地物分类
技术领域：
中的一种基于深度复卷积生成式对抗网络dccgan(deepcomplexconvolutionalgenerativeadversarialnetwork)模型的极化合成孔径雷达sar(syntheticapertureradar)图像分类方法。本发明可用于对极化sar图像中的地物目标进行分类，能有效的提高极化sar图像分类的精度，可用于目标识别与跟踪定位。
背景技术：
：极化sar是一种高分辨率主动式有源微波遥感成像雷达，具有全天候、全天时、分辨率高、可侧视成像等优点，能获得目标更丰富的信息。极化sar图像分类的目的是利用机载或者星载极化sar传感器获得的极化测量数据确定每个像素所属的类别，在农业、林业、军事、地质学、水文学和海洋等方面具有广泛的研究和应用价值。西安电子科技大学在其申请的专利文献“一种基于dcgan的极化sar图像分类方法”(专利申请号：201710440090.x，公开号：cn107292336a)中提出一种基于dcgan(deepconvolutionalgenerativeadversarialnetwork)的极化sar图像分类方法。该方法实现步骤为：(1)对极化数据进行pauli分解得到3个极化散射特征形成伪彩图；(2)对伪彩图进行取块组成基于图像块数据集；(3)将大量无标记数据集输入dcgan网络进行无监督训练，得到训练好的训练网络模型dcgan；(4)利用训练好的dcgan中的判别器d,更换判别器最后一层二分类器为softmax分类器，形成判别分类网络模型；(5)将少量有标记样本集输入判别分类网络模型进行有监督训练；(6)最后输入测试集到判别分类网络中进行分类。该方法存在的不足之处是，该方法通过pauli分解提取散射特征，未考虑到原始极化sar复数数据的丰富信息，忽略了图像某些相位信息，导致图像信息利用不充分，不能很好地将图像边缘检测出来，结果达不到很高的分类精度。liuf等人在其发表的论文“pol-sarimageclassificationbasedonwishartdbnandlocalspatialinformation”(ieeetransactionsongeoscience&remotesensing,2016,54(6):3292-3308.)中提出了一种wishartdbn(deep-belif-network)模型的极化sar分类方法。该方法首先进行极化目标分解从极化相干矩阵得到极化散射特征，然后将这些散射特征输入到dbn模型中进行高级特征提取，最后完成分类。其中，dbn能够完成从低层次到高层次的特征提取。dbn完成特征学习的前提，是假设数据服从高斯分布，但由极化目标分解提取的散射特征服从的是wishart分布。因此，将极化散射特征服从的wishart分布引入dbn，来完成特征学习，可以学习到更有效的特征，实现更好的分类效果。但是，该方法仍然存在的不足之处是，dbn不适于发掘图像里的局部空间关联信息，并且在初始化权值参数时随机初始化模型参数，导致模型训练速度变慢，很难收敛到全局最优解，得不到很高的准确率。技术实现要素：本发明的目的在于针对上述现有技术的不足，提出了一种基于深度复卷积生成式对抗网络dccgan模型的极化sar图像分类方法。本发明将已有的极化sar图像分类方法延拓至复数域进行运算，直接处理复数数据，充分利用极化sar数据的方向信息，更全面细致地描述图像特征，同时增强模型的泛化能力，从而有效地提高极化sar图像的分类精度。实现本发明的技术思路是：首先，对极化合成孔径雷达sar图像进行预处理并构造数据集；其次，搭建并训练深度复卷积生成式对抗网络dccgan模型；然后，搭建并训练判别分类网络模型；最后，用训练好的判别分类网络模型对极化合成孔径雷达sar图像进行分类。实现本发明目的的具体步骤如下：(1)输入一幅待分类的每一个像素点为2×2的极化散射矩阵的极化sar图像；(2)预处理数据：采用实虚分离方法对待分类的极化sar图像中的每一个像素点提取特征，得到极化sar图像的8维实数特征矩阵；(3)特征矩阵归一化及取块；将特征矩阵中的元素值归一化到[0，1]之间，得到归一化后的特征矩阵，对归一化后的特征矩阵按步长为1的滑窗取块，得到1560000个32×32的特征矩阵块，将所有特征矩阵块组成基于特征矩阵块的数据集；(4)构造数据集：(4a)在所有特征矩阵块中随机选取8％的特征矩阵块，组成无标签训练数据集；(4b)在有标记的特征矩阵块的每个类别中随机选取5％的特征矩阵块，组成有标签训练数据集；(4c)将选取有标签训练数据集后剩余的所有特征矩阵块，组成测试数据集；(5)构造深度复卷积生成式对抗网络dccgan模型：(5a)搭建一个依次由输入层→第一个复反卷积层→第二个复反卷积层→第三个复反卷积层→输出层组成的生成器g；(5b)搭建一个依次由输入层→第一个复卷积层→第二个复卷积层→第三个复卷积层→二分类器组成的判别器d；(5c)将生成器g和判别器d组成深度复卷积生成式对抗网络dccgan模型；(6)训练深度复卷积生成式对抗网络dccgan模型：将无标签训练数据集输入到深度复卷积生成式对抗网络dccgan模型中，对该模型进行训练，得到训练好的生成器和判别器组成的深度复卷积生成式对抗网络dccgan模型；(7)构造判别分类网络模型并初始化模型：(7a)构建一个依次由输入层→第一个复卷积层→第二个复卷积层→第三个复卷积层→softmax多分类器组成的判别分类网络模型；(7b)利用训练好的深度复卷积生成式对抗网络dccgan模型中判别器每一个复卷积层的权重，初始化判别分类网络模型对应复卷积层的参数，用(0,1)区间的随机数初始化softmax分类器的参数；(8)训练判别分类网络模型：(8a)将有标签训练数据集输入到判别分类网络模型中，更新softmax分类器参数，得到训练好的softmax分类器；(8b)将有标签训练数据集输入到判别分类网络模型中，更新整个判别分类网络模型参数，得到训练好的判别分类网络模型；(9)预测分类：利用训练好的判别分类网络模型，对待分类的极化sar图像进行分类，得到极化sar图像的像素类别。本发明与现有技术相比较，具有以下优点：第一，本发明构造了一个深度复卷积生成式对抗网络dccgan模型，通过该模型无需对极化合成孔径雷达sar图像的极化目标进行分解，可直接从极化散射矩阵中提取特征，克服了现有技术采用极化目标分解后再提取特征，导致的极化合成孔径雷达sar图像中某些相位信息被忽略，从而引起极化合成孔径雷达sar图像分类精度不高的问题，使得本发明具有充分利用极化合成孔径雷达sar图像的丰富信息的优点，有效地提高了极化合成孔径雷达sar图像的分类精度。第二，本发明利用训练好的深度复卷积生成式对抗网络dccgan模型的判别器，初始化判别分类网络模型的参数，克服了现有技术由于随机初始化模型参数，导致模型训练速度慢，难以收敛到全局最优解的问题，使得本发明对合成孔径雷达sar图像分类时加快网络模型的训练速度，快速收敛到全局最优解，提高了极化合成孔径雷达sar图像的分类效率。附图说明图1是本发明的流程图；图2是本发明的仿真图。具体实施方式下面结合附图对本发明做进一步的描述。参照图1，本发明实现的步骤如下：步骤1，输入一幅待分类的每一个像素点为2×2的极化散射矩阵的极化sar图像。步骤2，预处理数据。采用实虚分离方法对待分类的极化sar图像中的每一个像素点提取特征，得到极化sar图像的8维实数特征矩阵。从下式矩阵每一复数元素中分别提取回波数据的实数数值：其中，s表示待分类的极化合成孔径雷达sar图像每一个像素点的极化散射矩阵，[]表示矩阵符号，a表示输入的极化散射矩阵s中垂直方向接收的垂直方向发射的回波数据的实部的实数数值，b表示输入的极化散射矩阵s中垂直方向接收的垂直方向发射的回波数据的虚部的实数数值，i表示虚数单位，c表示输入的极化散射矩阵s中水平方向接收的垂直方向发射的回波数据的实部的实数数值，d表示输入的极化散射矩阵s中水平方向接收的垂直方向发射的回波数据的虚部的实数数值，e表示输入的极化散射矩阵s中垂直方向接收的水平方向发射的回波数据的实部的实数数值，f表示输入的极化散射矩阵s中垂直方向接收的水平方向发射的回波数据的虚部的实数数值，g表示输入的极化散射矩阵s中水平方向接收的水平方向发射的回波数据的实部的实数数值，h表示输入的极化散射矩阵s中水平方向接收的水平方向发射的回波数据的虚部的实数数值；将得到的a、b、c、d、e、f、g、h8个实数数值按a、c、e、g、b、d、f、h顺序排列，组成8维的实数特征矩阵。步骤3，特征矩阵归一化及取块。将特征矩阵中的元素值归一化到[0，1]之间，得到归一化后的特征矩阵。常用的归一化方法有：特征线性缩放法、特征标准化和特征白化。本实例采用特征线性缩放法，即先求出特征矩阵中所有元素值的最大值；再将特征矩阵中的每个元素值均除以最大值，得到归一化后的特征矩阵。对归一化后的特征矩阵按步长为1的滑窗取块，得到1560000个32×32的特征矩阵块，将所有特征矩阵块组成基于特征矩阵块的数据集。步骤4，构造数据集。在所有特征矩阵块中随机选取8％的特征矩阵块，组成无标签训练数据集。在有标记的特征矩阵块的每个类别中随机选取5％的特征矩阵块，组成有标签训练数据集。将选取有标签训练数据集后剩余的所有特征矩阵块组成测试数据集。步骤5，构造深度复卷积生成式对抗网络dccgan模型。第一步，搭建一个依次由输入层→第一个复反卷积层→第二个复反卷积层→第三个复反卷积层→输出层组成的生成器g，其中，每层的参数设置如下：输入层设置的单元个数为100；第一个复反卷积层设置的特征映射图数目为256，滤波器尺寸为5；第二个复反卷积层设置的特征映射图数目为128，滤波器尺寸为5；第三个复反卷积层设置的特征映射图数目为64，滤波器尺寸为5；输出层设置的特征映射图数目为8；第二步，搭建一个依次由输入层→第一个复卷积层→第二个复卷积层→第三个复卷积层→二分类器组成的判别器d，其中，每层的参数设置如下：输入层设置的特征映射图数目为8；第一个复卷积层设置的特征映射图数目为64，滤波器尺寸为5；第二个复卷积层设置的特征映射图数目为128，滤波器尺寸为5；第三个复卷积层设置的特征映射图数目为256，滤波器尺寸为5；二分类器设置的输出单元个数为1；第三步，将生成器g和判别器d组成深度复卷积生成式对抗网络dccgan模型。步骤6，训练深度复卷积生成式对抗网络dccgan模型。将无标签训练数据集输入到深度复卷积生成式对抗网络dccgan模型中，对该模型进行训练，得到训练好的生成器和判别器组成的深度复卷积生成式对抗网络dccgan模型。将100维的均匀噪声输入到生成器g中，将有标签训练数据集和生成器g的输出同时输入判别器d中，先更新判别器d的参数，再更新生成器g的参数，得到训练好的深度复卷积生成式对抗网络dccgan模型。步骤7，构造判别分类网络模型并初始化模型：构建一个依次由输入层→第一个复卷积层→第二个复卷积层→第三个复卷积层→softmax多分类器组成的判别分类网络模型，其中，每层具体参数设置如下：输入层设置的特征映射图数目为8；第一个复卷积层设置的特征映射图数目为64，滤波器尺寸为5；第二个复卷积层设置的特征映射图数目为128，滤波器尺寸为5；第三个复卷积层设置的特征映射图数目为256，滤波器尺寸为5；softmax多分类器设置的特征映射图数目为3。利用训练好的深度复卷积生成式对抗网络dccgan模型中判别器每一个复卷积层的权重，初始化判别分类网络模型对应复卷积层的参数，用(0,1)区间的随机数初始化softmax分类器的参数。步骤8，训练判别分类网络模型。将有标签训练数据集输入到判别分类网络模型中，更新softmax分类器参数，得到训练好的softmax分类器。训练softmax分类器的具体操作为：将有标签训练数据集输入到判别分类网络模型中，利用判别分类网络模型提取数据集中每个样本的特征，将属于人造目标的类别作为模型的输出，通过求解人造目标的类别与真实的地物标记的正确类别之间的误差并对误差反向传播，更新softmax分类器的参数，得到训练好的softmax分类器。将有标签训练数据集输入到判别分类网络模型中，更新整个判别分类网络模型参数，得到训练好的判别分类网络模型。训练判别分类网络模型的具体操作为：将有标签训练数据集输入到判别分类网络模型中，利用判别分类网络模型提取数据集中每个样本的特征，将属于人造目标的类别作为模型的输出，通过求解人造目标的类别与真实的地物标记的正确类别之间的误差并对误差进行反向传播，更新整个判别分类网络模型参数的参数，得到训练好的整个判别分类网络模型参数。步骤9，预测分类。利用训练好的判别分类网络模型，对待分类的极化sar图像进行分类，得到极化sar图像的像素类别。下面结合仿真图对本发明效果做进一步的说明：1、仿真实验条件：本发明的仿真实验是在主频2.40ghz*16的intel(r)xeon(r)e5-2630cpu、内存64gb的硬件环境和tensorflow的软件环境下进行的。图2(a)是本发明的仿真实验所使用的待分类的极化sar图像，该图为德国dlr的esar传感器在德国慕尼黑附近的oberpfaffenhofen小镇获取的l波段全极化数据，图像大小为1300×1200，总像素点个数为1560000个，有标签像素点个数为1374298个。图2(b)是本发明的仿真实验中用于比对的极化合成孔径雷达sar真实的地物标记图。图2(b)中的白色代表背景，黑色代表类别1，深灰色代表类别2，浅灰色代表类别3。2、仿真结果分析：将分类结果的像素类别与真实地物类别进行对比，将类别一致的像素个数与全部像素个数的比值作为极化sar图像的总准确率，在每一类别中将类别一致的像素个数与该类别的像素个数的比值作为每一类别的分类准确率。本发明的分类总准确率为97.4187％，本发明分类结果如图2(c)所示。在相同仿真实验条件下，将现有技术的基于深度卷积生成式对抗网dcgan模型的极化sar分类方法进行仿真实验，得到对比实验分类结果图如图2(d)所示。将本发明和对比实验基于深度卷积生成式对抗网dcgan模型的极化sar分类方法的每一类别的准确率进行比较，结果如表1所示。表1分类方法dcgan本发明类别1(％)94.888295.6611类别2(％)95.377496.4128类别3(％)98.417698.6950总准确率96.952197.4998由表1可见，本发明的每一类别的准确率均比深度卷积生成式对抗网dcgan模型的极化sar分类方法的分类效果好，提高了分类精度。对比图2(b)可知，本发明的结果图中分类结果的区域一致性较好，不同区域划分后的边缘清晰可辨，且保持了细节信息，分类结果图中杂点也比较少。以上实验表明，本发明通过深度复卷积生成式对抗网络dccgan模型对极化合成孔径雷达sar图像进行分类，无需对极化合成孔径雷达sar图像的极化目标进行分解，可直接从极化散射矩阵中提取特征，充分利用极化sar数据的方向信息，更全面细致地描述图像特征，有效地提高了极化合成孔径雷达sar图像的分类精度。当前第1页12