[发明专利]遥感图像舰船检测方法、装置、设备及其存储介质

权利要求书

1.一种基于深度学习的遥感图像舰船检测方法，其特征在于，该方法包括：

获取待检测的遥感图像；

对所述遥感图像进行处理得到输入图像；

将所述输入图像输入到预先构建的目标检测模型得到目标检测结果，所述目标检测模型是基于深度学习算法训练得到的，所述目标检测结果至少包括预测目标物为舰船的条件概率和所述预测目标物为舰船的置信度。

2.根据权利要求1所述的方法，其特征在于，所述对所述遥感图像进行处理至少包括：

对所述遥感图像进行滑动窗口处理得到输入图像集合，所述输入图像集合包括所述滑动窗口与所述遥感图像具有重叠区域的所有子图像。

3.根据权利要求2所述的方法，其特征在于，对所述遥感图像进行滑动窗口处理得到输入图像集合，包括：

将所述滑动窗口按照预设步长在所述遥感图像上移动，所述预设步长满足小于所述滑动窗口的条件；

截取所述滑动窗口与所述遥感图像具有重叠区域的图像部分作为所述子图像。

4.根据权利要求3所述的方法，其特征在于，该方法还包括：

根据所述置信度与第一预设阈值的比较结果，输出所述预测目标物为舰船的第一候选框集合；

对所述第一候选框集合进行第一非极大值抑制处理，得到第二候选框集合。

5.根据权利要求4所述的方法，其特征在于，在得到第二候选框集合之后，该方法还包括：

对所述第二候选框集合进行第二非极大值抑制处理，得到最后的检测结果。

6.根据权利要求5所述的方法，其特征在于，对所述第二候选框集合进行第二非极大值抑制处理，包括：

确定所述第二候选框集合中所述置信度的最大值所对应的候选框为比较对象；

选择所述第二候选框集合中任一个其他候选框作为第二候选框，所述其他候选框不包括所述比较对象；

计算所述第二候选框与所述比较对象的交集面积和并集面积的比值；

将所述比值与第二预设阈值进行比较；

若所述比值超过所述第二预设阈值，则删除所述第二候选框；

若所述比值低于所述第二预设阈值，则保留所述第二候选框；

重复上述步骤直到确定所述第二候选框集合中唯一的最佳候选框，作为最终的检测结果。

7.根据权利要求6所述的方法，其特征在于，该方法还包括构建目标检测模型的步骤，该步骤包括：

获取预设数值的历史遥感图像；

对所述历史遥感图像进行预处理；

将预处理后的结果划分为训练集和验证集；

按照YOLOv2算法对所述训练集进行训练学习得到所述目标检测模型；以及

利用所述验证集确定所述目标检测模型的参数。

8.根据权利要求7所述的方法，其特征在于，所述预处理至少包括：

基于所述历史遥感图像确定样本图像集合；

对所述样本图像集合中每一张样本图像进行标注，生成标签文件；

对所述样本图像集合中的每一张样本图像进行角度旋转处理；

对所述标签文件中每一张图像按照变换公式进行角度旋转处理。

9.一种基于深度学习的遥感图像舰船检测装置，其特征在于，该装置包括：

图像获取单元，用于获取待检测的遥感图像；

图像处理单元，用于对所述遥感图像进行处理得到输入图像；

目标检测单元，用于将所述输入图像输入到预先构建的目标检测模型得到目标检测结果，所述目标检测模型是基于深度学习算法训练得到的，所述目标检测结果至少包括预测目标物为舰船的条件概率和所述预测目标物为舰船的置信度。

10.根据权利要求9所述的装置，其特征在于，所述图像处理单元至少包括：

滑动处理子单元，用于对所述遥感图像进行滑动窗口处理得到输入图像集合，所述输入图像集合包括所述滑动窗口与所述遥感图像具有重叠区域的所有子图像。