一种基于稀疏SAR幅度-相位图像数据集的目标分类方法

摘要

本发明公开了一种基于稀疏SAR幅度‑相位图像数据集的目标分类方法，首先，基于匹配滤波复图像利用BiIST算法重构稀疏SAR图像；其次，利用稀疏SAR图像制作幅度‑相位双通道的图像数据集；最后，将待分类的稀疏SAR图像数据集作为幅度‑相位卷积神经网络的输入，输出得到目标分类的结果及准确率。本发明采用的基于复图像的稀疏重构算法不仅可以降低系统计算复杂度，而且可以在提升图像质量的同时保留图像的相位信息，为后续幅度‑相位卷积神经网络的训练提供保障；本发明提出的幅度‑相位卷积神经网络将SAR图像的幅度和相位作为训练网络的一个双通道输入，同时利用了图像的幅度和相位信息，可以实现目标分类精度的进一步提升。

说明书

技术领域

本发明属于雷达图像处理和目标分类领域，尤其涉及一种基于稀疏SAR幅度-相位图像数据集的目标分类方法。

背景技术

合成孔径雷达(Synthetic Aperture Radar，SAR)作为一种主动式微波遥感，可以不受时间和天气的影响，实现全天时全天候的对地观测，目前被广泛地用于军事和民用领域。2006年Hinton教授提出了深度学习的概念，指出多层的卷积神经网络具有强大的特征学习的能力，对于分类方面的问题有着很重要的研究意义。AlexNet模型是由Krizhevesky等人在2012年提出的一个深度卷积神经网络(Convolutional Neural Networks，简称CNN)模型，获得了ImageNet比赛的冠军，在图像分类方面取得了突破性的进展。考虑到光学图像和SAR图像的相似性，CNN同样可以适用于SAR图像的目标分类。针对SAR图像中同时具有幅度和相位两种信息的情况，不少学者也展开了一系列的研究。在2017年，Zhang等人提出了一种复值CNN，并推导了基于随机梯度下降的复反向传播算法，文中极化SAR数据的分类实验结果表明复值CNN相较于传统的CNN可以进一步地降低分类误差。2018年，Coman等人为了将相位信息纳入到深度学习的训练学习过程中，采用了由幅度-实部-虚部三层数据构成输入图像的方法，基于MSTAR数据集的实验证实，相较于传统的幅值网络，该方法可以有效的提高分类精度，并且在一定程度上解决了训练数据缺乏造成的过拟合问题。

由于匹配滤波复图像的质量通常不高，存在比较严重的杂波和旁瓣，而BiIST算法相较于传统的稀疏重构算法，其得到的非稀疏解不仅可以实现图像质量的提升，而且可以保留SAR图像的相位信息，为后续基于幅度和相位信息的分类网络的训练提供了保障。

发明内容

发明目的：本发明的目的是提供一种基于稀疏SAR图像的幅度和相位信息进行训练的全新的幅度-相位卷积神经网络，用于目标分类，提升分类精度。

技术方案：本发明提供了一种基于稀疏SAR幅度-相位图像数据集的目标分类方法，包括以下步骤：

(1)基于匹配滤波复图像，采用BiIST算法对复图像进行重建，得到具有保留图像相位信息的稀疏SAR图像；

(2)将稀疏SAR图像的幅度和相位作为网络的两个通道输入，制作并划分出训练集、测试集和验证集；

(3)将步骤(2)得到的待分类的稀疏SAR图像数据集作为幅度-相位卷积神经网络的输入，输出得到目标分类的结果及准确率。

进一步地，所述步骤(1)包括以下步骤：

(11)第i+1步观测场景的非稀疏估计为：

其中，

(12)第i+1步的正则化参数β

其中，

(13)第i+1步的中间变量W

(14)第i+1步观测场景的稀疏估计为：

其中，

(15)第i+1步的迭代误差表示为：

当迭代误差满足条件Residual≤ε时，结束循环，分别输出场景的稀疏重建结果

和

否则i＝i+1，继续执行步骤(11)-(15)中的循环过程，ε表示重建误差参数。

进一步地，步骤(2)所述训练集包含对应的目标类别标签。

进一步地，步骤(2)训练集、测试集和验证集的比例为7∶2∶1。

进一步地，步骤(3)所述的幅度-相位卷积神经网络包括4个卷积层，3个平均池化层，2个全连接层以及softmax分类层；卷积层根据图像在空间上的局部相关性自动提取特征，然后利用池化层对提取的特征进行采样，最后全连接层将所有的特征图连接起来通过softmax分类器获得分类结果。

有益效果：与现有技术相比，本发明的有益效果：1、相比于传统的正则化稀疏重构算法，BiIST算法可以同时得到观测场景的稀疏估计和非稀疏估计，降低系统的计算复杂度，其中的非稀疏估计不仅可以实现图像质量的提升，而且可以保留图像的相位信息；2、本发明中提出的幅度-相位卷积神经网络相较于目前用于目标分类的主流卷积神经网络同时利用了SAR图像的幅度信息和相位信息，在一定程度上可以提升目标的分类精度。

附图说明

图1为本发明的流程图；

图2为本发明中提出的幅度-相位卷积神经网络结构示意图；

图3为本发明中提出的幅度-相位卷积神经网络的训练流程图；

图4为本发明重建的稀疏图像数据集分别在幅度卷积神经网络和幅度-相位卷积神经网络下的平均分类精度曲线图。

具体实施方式

下面结合附图对本发明做进一步详细说明。

本发明提供一种基于稀疏SAR幅度-相位图像数据集的目标分类方法，如图1所示，包括如下步骤：

步骤1：基于匹配滤波复图像利用BiIST算法重构稀疏SAR图像。

本发明基于匹配滤波复图像，采用BiIST算法对复图像进行重建，得到具有保留图像相位信息的稀疏SAR图像。

以第i+1步迭代为例，BiIST算法的迭代过程可以表示为：

1)第i+1步观测场景的非稀疏估计为：

其中，

2)第i+1步的正则化参数β

其中，

3)第i+1步的中间变量W

4)第i+1步观测场景的稀疏估计为：

其中，

5)第i+1步的迭代误差表示为：

当迭代误差满足条件Residual≤ε时，结束循环，分别输出场景的稀疏重建结果

和

否则i＝i+1，继续执行步骤S11-S15中的循环过程；ε表示重建误差参数，在本发明中选择其值为10

其中，非稀疏重建结果

步骤2：利用步骤1得到的稀疏SAR图像制作幅度-相位双通道的图像数据集。

将得到的稀疏SAR图像的幅度和相位作为网络的两个通道输入，制作并划分出训练集、测试集和验证集，训练集、测试集和验证集的比例为7∶2∶1；训练集应包含对应的目标类别标签。

步骤3：将步骤2得到的待分类的稀疏SAR图像数据集作为幅度-相位卷积神经网络的输入，输出得到目标分类的结果及准确率。

幅度-相位卷积神经网络的模型如图2所示，由4个卷积层，3个平均池化层，2个全连接层以及softmax分类层构成。如图3所示，输入是由幅度层和相位层构成的双通道数据，卷积层可以根据图像在空间上的局部相关性自动提取特征，然后利用池化层对提取的特征进行采样，最后全连接层将所有的特征图连接起来通过softmax分类器获得分类结果。

将训练集作为提出的幅度-相位卷积神经网络的输入进行训练，将待分类的SAR图像数据集输入训练好的网络，得到目标的分类结果及分类精度。

以公开的MSTAR的十类目标数据集为例对本发明提出的一种基于稀疏SAR幅度-相位图像数据集的目标分类方法进行验证。

表1为重建的稀疏SAR图像经过幅度-相位卷积神经网络的目标分类结果

表1为本发明重建的稀疏图像数据集经过幅度-相位卷积神经网络的目标分类结果。BiIST算法相较于传统的稀疏重构算法可以得到具有保相性的稀疏图像，将重构得到的稀疏SAR图像应用于本发明提出的目标分类网络，将幅度和相位信息同时作为输入数据进行训练，将待分类的数据集输入训练好的网络，可以得到混淆矩阵，包括目标的分类结果以及分类精度。图4表示本发明重建的稀疏图像数据集分别在幅度卷积神经网络和幅度-相位卷积神经网络下的平均分类精度曲线。采用了样本数为1000、1200、1500、1700和2000的训练集分别在幅度卷积神经网络和幅度-相位卷积神经网络进行训练。结果表明，相较于仅利用幅度信息进行训练的传统卷积神经网络，本发明提出的幅度-相位卷积神经网络的目标分类精度要更高，并且在样本数量较少的情况下，表现出更好的分类性能。