[发明专利]高分辨率卫星影像任意方向舰船目标检测方法

权利要求书

1.一种高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，包括：

将卫星影像输入到区域筛选网络，得到舰船目标候选区域图像；

将舰船目标候选区域图像输入到预先训练好的任意方向舰船检测模型中进行目标检测，得到初步目标检测结果；

利用NMS方法对初步目标检测结果进行去重，得到目标检测结果；

其中，任意方向舰船检测模型采用VGG16作为骨干网络，保留conv1_1至conv5_3，将最后两层全连接层替换为卷积层，并依次增加conv6_1、conv6_2，conv7_1、conv7_2，conv8_1、conv8_2和conv9_1、conv9_2共8层卷积层以及一个3x5的长卷积核生成的卷积层。

2.根据权利要求1所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，任意方向舰船检测模型的训练方法具体包括：

训练数据生成步骤：将预先标注好的高分辨率卫星影像裁剪成固定大小的样本数据；

数据增广步骤：基于样本数据和超分辨率对抗生成网络生成多尺度高分辨率影像，作为训练数据；采用旋转、翻转和亮度对比度调整方法进一步扩充样本数据，作为训练数据；

长宽比聚类分析步骤：对样本数据进行舰船目标的长、宽以及长宽比聚类分析，得到聚类结果，根据聚类结果设定长宽比参数；

训练步骤：基于长宽比参数和预设的默认框垂直偏移，将训练数据输入到任意方向舰船检测模型中进行训练，并利用NMS算法进行去重，在训练过程中基于损失函数进行收敛判断，最终得到训练好的任意方向舰船检测模型。

3.根据权利要求2所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，训练数据生成步骤具体包括：

将深度学习网络输入图像大小作为滑动窗口的大小，有重叠的在卫星影像上滑动；

若当前滑动窗口内包含有效舰船目标，则将当前滑动窗口对应的图像裁剪出来，同时，将当前窗口内舰船相对于整幅卫星影像的坐标更新为在当前滑动窗口内的坐标，并保存为裁剪出的图像对应的XML标注文件。

4.根据权利要求3所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，有效舰船目标的判别方法包括：

滑动窗口面积和舰船目标四边形区域重叠面积与舰船四边形区域面积之比大于0.5，则判定为有效舰船目标。

5.根据权利要求2所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，训练步骤中所采用的损失函数为：

其中，x表示匹配指示矩阵，对于第i个默认框和第j个真实目标框，若二者匹配，则x_ij＝1，c表示预测的置信度，l表示预测的位置，g表示真实的位置；N为与真实目标框匹配的默认框的总数，L_loc为L1损失，L_conf为二值的softmax分类损失。

6.根据权利要求1所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，所述区域筛选网络采用MTCNN模型中的PNet网络。

7.根据权利要求2～6任意一项所述的高分辨率卫星影像任意方向舰船目标检测方法，其特征在于，多尺度高分辨率影像包括：原始影像2倍分辨率的影像数据和原始影像4倍分辨率的影像数据。