[发明专利]一种基于迁移学习的SAR遥感图像舰船检测方法

摘要

本发明涉及一种基于迁移学习的SAR遥感图像舰船检测方法，主要解决现有的SAR遥感图像没有舰船的标注，不能直接将深度学习应用于SAR遥感图像舰船检测的问题。本发明的具体步骤如下：(1)准备相关数据集，包括光学遥感图像舰船数据集和SAR遥感图像数据集；(2)基于迁移学习的思想，设计相应的网络结构和损失函数，同时利用光学遥感图像和SAR遥感图像数据集进行网络的训练；(3)通过训练出来的模型，对SAR遥感图像进行舰船检测。利用光学遥感图像和SAR遥感图像中舰船的共性，即使模型训练时没有SAR遥感图像中舰船的标注，仍可以对SAR遥感图像中的舰船进行检测。

主权项

1.一种基于迁移学习的SAR遥感图像舰船检测方法，其特征在于，包括以下步骤：步骤1：准备相关数据集，包括光学遥感图像舰船数据集作为源域和SAR遥感图像数据集作为目标域，其中光学遥感图像舰船数据集有舰船标注，SAR遥感图像数据集没有舰船标注；步骤2：基于迁移学习的思想，设计用于训练的网络和损失函数；步骤2.1：设计目标域与源域的私有编码结构和共享编码结构，以VGG16作为基础结构，将全连接层FC6和FC7变成卷积层，同时，额外增加两个卷积层以及一个全连接层，每个卷积层后面跟随BN操作，全连接层的输出大小为100；步骤2.2：设计损失函数，包括相似损失函数，区别损失函数，以及分类损失和回归损失；其中相似损失函数，完成两个领域的共享表示；区别损失函数，保证每个域的独立特性；分类和回归损失用于舰船的检测；相似损失函数；将目标域与源域共享编码结构得到的目标域图像特征与源域图像特征，通过梯度反转层和领域分类器实现目标域图像特征与源域图像特征最大程度的“混淆”，完成领域的共享；具体的公式为：其中d,0和1分别代表源域和目标域，d是已知的，则是通过目标域和源域共享编码结构得到的特征作为依据预测出来的，通过梯度反转层将该损失最大化；梯度反转层在原函数梯度前加一个负号，达到仍然是求损失的最小值的目的；区别损失函数；通过私有和共享领域特征之间的软子空间正交性约束定义，使共享和私有编码去编码输入的不同方面；该损失函数的输入为目标域私有编码结构得到的目标域图像特征和源域私有编码结构得到的源域图像特征；具体的公式为：其中是源域数据通过共享编码结构得到的特征矩阵，是源域数据通过私有编码结构得到的特征矩阵，是目标域数据通过共享编码结构得到的特征矩阵，是目标域数据通过私有编码结构得到的特征矩阵；分类和回归损失，其中分类损失采用softmax loss，回归损失采用smoothL1 loss；具体的公式为：分类和回归损失是用在检测结果上的，在训练过程中，是用带标签的源域数据通过上述损失来训练用于舰船检测的网络；步骤3：使用设计的网络对数据集进行训练，得到用于SAR遥感图像舰船检测的模型；步骤3.1：将源域和目标域数据同时输入到网络中；选取源域遥感图像和目标域遥感图像，将他们均值中心化到一定范围，每张均值中心化后的源域遥感图像数据表示为xs，每张均值中心化后的目标域遥感图像数据表示为xt，将xs和xt其输入到网络中；步骤3.2：使用私有编码结构提取私有特征，共享编码结构来提取共有特征；私有编码结构和共享编码结构采用的是修改后的VGG16网络，网络的初始参数设置为在ImageNet数据库上预训练的VGG16模型参数；通过私有编码结构，输入的源域数据x^t和目标域数据x^t将分别得到源域的私有特征和目标域私有特征同理，通过共享编码结构，输入的源域数据x^s和目标域数据x^t将分别得到源域的共有特征和目标域的共有特征步骤3.3：通过提取出的特征，计算出相似损失函数值和区别损失函数的值；步骤3.3.1：将得到的共有特征和加上一个梯度反转层，保证在计算梯度时会加上负号，再处理后的共有特征和输入到一个二分类器中，判断输入的特征属于源域还是目标域，输出为其中0和1分别代表源域和目标域；通过所述相似损失函数计算公式计算相似损失函数值；步骤3.3.2：将得到的源域私有特征目标域私有特征源域共有特征和目标域共有特征构建成相应的矩阵矩阵每行即为每张图像得到的相应特征，在根据所述区别损失函数计算公式计算区别损失函数值；步骤3.4：选取源域所提取的共有特征，使用SSD结构进行检测，结合源域标注，得到分类和回归损失函数的值；使用SSD结构进行检测，通过检测结果和已知的源域数据标注，结合所述分类和回归损失函数计算公式计算出相应的分类和回归损失函数值；步骤3.5：结合所有损失值，利用反向传播算法对网络进行调整：通过之前步骤计算出来的各个损失值，将各个损失值加上系数进行累加，得到总体损失值，具体公式为：其中α，β，γ，为各个损失的系数，用来控制各个损失值的权值；通过得到的总体损失值利用反向传播算法对网络整体进行训练；步骤4：使用得到的SAR遥感图像舰船检测模型，进行舰船检测：取SAR遥感图像，将其进行均值中心化，得到输入x；将得到的x输入到已训练的模型中；利用模型中训练好的共享编码结构提取出输入x的共享编码特征hc；将提取出的共享编码特征hc输入到SSD检测框架下，进行舰船检测。