一种基于迁移学习的SAR目标识别系统及方法技术方案

本发明专利技术提供一种深度学习与迁移学习结合的SAR目标识别方法，采用的方案可以分为四个步骤：一、准备阶段：读取预训练模型，并读取图片目录，划分训练集、验证集及测试集；二、训练阶段：构造深度学习与迁移学习结合的全连接神经网络分类器，并将图片集作为预训练模型的输入，使用输出来更新全连接神经网络分类器的参数；三、存储阶段：存储模型结果；四、用MATLAB GUI图形用户界面设计一套目标背景图像合成并识别的图像处理系统。以上步骤的组合可以达到快速便捷的扩展数据集的样本数量并能实时显示出识别结果，提高了工作效率，为使用者提供相对更准确的识别结果。

A SAR Target Recognition System and Method Based on Transfer Learning

【技术实现步骤摘要】
一种基于迁移学习的SAR目标识别系统及方法
本专利技术涉及图像识别领域，特别是一种基于迁移学习的SAR目标识别系统及方法，该方法针对陆战场中地物目标SAR(SyntheticApertureRadar，合成孔径雷达)图像的一种快速检测并识别的方法。
技术介绍
近年来，随着机器学习的发展和SAR图像分辨率的不断提高，遥感技术被广泛的应用在地物目标的检测和识别工作中，对于SAR图像信息的解译及图像目标的识别要求也越来越高。传统的检测识别方法一般由SAR图像的噪声抑制、增强、边缘检测、目标图像分割、目标检测和目标识别几部分组成，然而这种方法的不足之处在于：1、实地考察获取样本标签方式受限于空间范围和地区的可访问性，代价昂贵且获取样本数量较少。2、每出现新的目标图像都需要重新进行带标签样本采集，增加了分类成本。3、步骤繁琐，需人工按步骤操作，效率低下。
技术实现思路
本专利技术提供一种深度学习与迁移学习结合的SAR目标识别方法，以克服现有技术存在的操作复杂，效率低下，样本数量少，获取包含目标的场景图像成本高及检测识别准确性低下等问题。为了达到本专利技术的目的，本专利技术提出的方案如下：一种基于迁移学习的SAR目标识别系统及方法，包括如下步骤：步骤1、准备阶段：读取预训练模型，并读取图片目录，划分训练集、验证集及测试集；步骤2、训练阶段：构造深度学习与迁移学习结合的全连接神经网络分类器，并将图片集作为预训练模型的输入，使用输出来更新全连接神经网络分类器的参数；步骤3、存储阶段：存储模型结果；步骤4、用MATLABGUI图形用户界面设计目标背景图像合成并识别的图像处理系统。进一步的，所述步骤1具体包括如下步骤：步骤101、数据集的制作及划分；步骤201、对场景图像和目标图像选择Poisson图像编辑算法进行无缝镶嵌，建立包含目标的场景图像测试集。进一步的，所述步骤2具体包括如下步骤：步骤201、选择预训练模型；步骤202、利用迁移学习在相似数据集上的局部特征共享性对网络进行初始化，并用它在目标数据集上进行参数微调；步骤203、构造全连接神经网络分类器进行数据的训练和参数的更新；步骤204、针对上述操作存在的复杂度高以及效率低下等问题设计了一套整合了上述步骤且可独立操作的系统。进一步的，所述步骤4的MATLABGUI图形用户界面具体功能分为：目标图像读取及处理部分；背景图像读取及目标-背景图像合成部分；合成后图像识别及结果显示部分。进一步的，训练阶段，网络模型选用fasterrcnn模型为基础进行数据的训练；框架选取Tensorflow框架。与现有技术相比，本专利技术的有益效果是：1)所述基于深度学习与迁移学习的图像识别方法，能在有限的训练时间和训练样本数量的基础上，为使用者提供相对更准确的识别结果。2)在训练阶段，基于本专利技术设计的独特的网络模型结构，训练数据集需求量大大缩小，深度神经网络通常需要上万张图像用于训练，本模型仅仅使用约500张原始图像用于训练。3)本专利技术填补了目前SAR目标识别研究没有包含目标的场景数据集的空白，可快速构建包含目标的场景SAR图像数据集，且预训练结果可用于其他各种复杂环境中。可同时识别至少3种不同种类的SAR目标且识别准确率达到98％以上。附图说明图1为本专利技术SAR目标识别系统功能框图；图2为本专利技术SAR目标识别系统总体设计框图；图3为对数据集中目标及场景图像进行图像合成算法流程图；图4为结合了迁移学习方法的算法结构框图；图5为对原数据集采用不同方式进行扩充对比图；图6为合成后的场景图像示意图；图7为本专利技术SAR目标识别系统界面示意图；图8为本专利技术SAR目标识别系统运行界面图。具体实施方式下面通过具体实施方式结合附图对本专利技术作进一步详细说明。其中不同实施方式中类似元件采用了相关联的类似的元件标号。在以下的实施方式中，很多细节描述是为了使得本申请能被更好的理解。然而，本领域技术人员可以毫不费力的认识到，其中部分特征在不同情况下是可以省略的，或者可以由其他元件、材料、方法所替代。在某些情况下，本申请相关的一些操作并没有在说明书中显示或者描述，这是为了避免本申请的核心部分被过多的描述所淹没，而对于本领域技术人员而言，详细描述这些相关操作并不是必要的，他们根据说明书中的描述以及本领域的一般技术知识即可完整了解相关操作。本专利技术提供一种深度学习与迁移学习结合的SAR目标识别方法，采用的方案可以分为四个步骤：一、准备阶段：读取预训练模型，并读取图片目录，划分训练集、验证集及测试集；二、训练阶段：构造深度学习与迁移学习结合的全连接神经网络分类器，并将图片集作为预训练模型的输入，使用输出来更新全连接神经网络分类器的参数；三、存储阶段：存储模型结果；四、用MATLABGUI图形用户界面设计一套目标背景图像合成并识别的图像处理系统。以上四个步骤的组合可以达到可以快速便捷的扩展数据集的样本数量并能实时显示出识别结果，提高了工作效率，为后续工作奠定了基础。优选地，采用Poisson图像编辑算法进行SAR图像合成，以达到扩大数据集的数量，降低数据获取成本的目的。准备阶段具体包括如下步骤：1、数据集的制作及划分；2、对场景图像和目标图像选择Poisson图像编辑算法进行无缝镶嵌，建立包含目标的场景图像测试集。这种方法和传统方法对比它的优势在于：选择融合区域的过程简单且方便，而且最终可以得到无缝融合的结果。优选地，采用深度学习与迁移学习结合的方法针对SAR数据集进行训练，以达到提高分类速度及分类准确率的目的。训练阶段具体包括如下步骤：1、选择合适的预训练模型；2、利用迁移学习在相似数据集上的局部特征共享性对网络进行初始化，并用它在目标数据集上进行参数微调；3、构造全连接神经网络分类器进行数据的训练和参数的更新；4、针对上述操作存在的复杂度高以及效率低下等问题设计了一套整合了上述步骤且可独立操作的系统。优选地，采用MATLAB进行目标识别系统的GUI界面设计，以达到对数据集及识别工作的便捷操作、提高工作效率的目的。系统功能分为以下几个部分：1、目标图像读取及处理部分；2、背景图像读取及目标-背景图像合成部分；3、合成后图像识别及结果显示部分。其中，在采用深度学习与迁移学习结合的方法进行数据集的训练阶段，在网络模型的选取方面，选用2016年KaimingHe和ShaoqingRen合作提出的fasterrcnn模型为基础进行数据的训练，在结构上，fasterrcnn将特征提取，候选区域提取及分类都整合在了一个网络中，提高了综合性能有较大，在检测速度方面尤为显著。在框架选取方面，选用现在最流行的Tensorflow框架，它支持多种开发语言和操作系统，并且学习社区文档资料健全，目前，Tensorflow是使用者最多、最受欢迎的开源深度学习框架。实施例：由附图1可知，本专利技术实施例中提供了一种基于深度学习与迁移学习的SAR目标识别方法，该方法包括下述步骤：读取图片目录，对原始数据集中的单目标图像进行旋转、平移、缩放、镜像、转置五种不同的变换处理制作训练集及测试集，利用迁移学习的思想对所选网络模型进行预训练，保留初始网络参数，读取预训练模型，构造全连接神经网络分类器，加入剩余的样本进行训练，更新分类器参数，形成已训练的网络，在测试集部分利用本文档来自技高网...

【技术保护点】
1.一种基于迁移学习的SAR目标识别系统及方法，其特征在于，包括如下步骤：步骤1、准备阶段：读取预训练模型，并读取图片目录，划分训练集、验证集及测试集；步骤2、训练阶段：构造深度学习与迁移学习结合的全连接神经网络分类器，并将图片集作为预训练模型的输入，使用输出来更新全连接神经网络分类器的参数；步骤3、存储阶段：存储模型结果；步骤4、用MATLAB GUI图形用户界面设计目标背景图像合成并识别的图像处理系统。

【技术特征摘要】
1.一种基于迁移学习的SAR目标识别系统及方法，其特征在于，包括如下步骤：步骤1、准备阶段：读取预训练模型，并读取图片目录，划分训练集、验证集及测试集；步骤2、训练阶段：构造深度学习与迁移学习结合的全连接神经网络分类器，并将图片集作为预训练模型的输入，使用输出来更新全连接神经网络分类器的参数；步骤3、存储阶段：存储模型结果；步骤4、用MATLABGUI图形用户界面设计目标背景图像合成并识别的图像处理系统。2.根据权利要求1所述基于迁移学习的SAR目标识别系统及方法，其特征在于，所述步骤1具体包括如下步骤：步骤101、数据集的制作及划分；步骤201、对场景图像和目标图像选择Poisson图像编辑算法进行无缝镶嵌，建立包含目标的场景图像测试集。3.根据权利要求1所述基于迁移学习的SAR目标识别系统及方法，其特征在于，...

【专利技术属性】
技术研发人员：邸若海，乔梦雨，孙梦宇，闫芬婷，王柄凯，张宽，李亮亮，张颖，付博雯，
申请(专利权)人：西安工业大学，
类型：发明
国别省市：陕西,61