[发明专利]一种基于深度学习的遥感图像水面潜艇识别方法

说明书

本发明提供一种基于深度学习的遥感图像水面潜艇识别方法，首先建立潜艇遥感图像样本库；对所述图像样本库中的各潜艇遥感图像分别进行图像增强，得到增强后的图像样本库；其次，对所述增强后的图像样本库中的各潜艇遥感图像中的潜艇分别进行矩形框标注，得到标注数据集；利用yolov3目标检测网络，融合语义分割与目标检测所得到的特征，得到分割预测融合特征；根据所述标注数据集和所述分割预测融合特征，得到潜艇检测识别模型；利用所述潜艇识别模型进行潜艇识别，并输出潜艇位置信息和置信度。本发明的有益效果：解决背景与目标颜色相近的潜艇水面遥感图像的识别，提升了潜艇目标检测定位和识别的精度。

技术领域

本发明涉及图像处理领域，特别涉及一种基于深度学习的遥感图像水面潜艇识别方法。

背景技术

深度学习在图像识别领域得到广泛应用，使得识别精度得到了较大提升。以深度学习为基础，将深度学习目标识别算法应用到对遥感图像水面潜艇识别。

近年，对军事目标检测算法的研究，取得许多成果。专利文献1(CN 107844750A)，提出了一种水面全景图像目标检测识别方法，该方法首先利用目标检测识别神经网络获取目标类别和目标位置，然后对局部区域中目标进行精细分割定位，从而在全景图像中将目标位置准确识别出来。专利文献2(CN 108596030 A)，提出了基于Faster R-CNN的声纳目标检测方法，该方法首先采用深度学习技术对不同复杂水下环境中的目标进行特征提取，然后创建基于区域加速卷积神经网络(Fast-RCNN)声纳识别网络对声纳数据进行目标检测。专利文献3(CN 108319949A)，提出了一种高分辨率遥感图像中多朝向舰船目标检测与识别方法，该方法首先对舰船目标标注多角度矩形框，以及对数据增广处理，然后利用深度学习网络模型训练出舰船目标检测和识别模型，实现小样本环境下的高分辨率遥感图像中的多朝向、不同尺度和类型的舰船目标的快速精确检测识别。

但是，专利文献1中公开的水面全景图像目标检测识别方法尚未涉及背景较为复杂的卫星图像的识别；专利文献2中公开的基于Faster R-CNN的声纳目标检测方法也仅仅是利用声纳数据训练深度学习识别模型；专利3中公开的方法能对遥感图像中多朝向舰船目标进行检测与识别，但未涉及背景与目标颜色相近的潜艇水面遥感图像的识别。

发明内容

为了解决上述问题，本发明提供了一种基于深度学习的遥感图像水面潜艇识别方法。通过语义分割网络，与yolov3的特征提取网络同步融合从而快速准确提取潜艇特征，通过深度学习网络训练改进yolov3目标检测算法模型训练检测识别模型实现对潜艇准确检测与识别，减少漏检和误检。种基于深度学习的遥感图像水面潜艇识别方法，具体步骤如下：

S101：建立潜艇遥感图像样本库；

S102：对所述图像样本库中的各潜艇遥感图像分别进行图像增强，得到增强后的图像样本库；

S103：对所述增强后的图像样本库中的各潜艇遥感图像中的潜艇分别进行矩形框标注，进而得到所述增强后的样本图像库中所有的潜艇遥感图像所对应的标注数据集；

S104：利用yolov3目标检测网络，融合语义分割与目标检测所得到的特征，得到分割预测融合特征；

S105：根据所述标注数据集和所述分割预测融合特征，得到潜艇检测识别模型；

S106：采集潜艇实时图像，并进行图像增强，进而将增强后的潜艇实时图像输入所述潜艇检测识别模型进行潜艇识别；

S107：设置特征阈值α，并判断图像特征值是否超过特征阈值α？若是，则输出潜艇识别预测结果，程序结束；否则，返回至步骤S106。

进一步地，在步骤S101中，建立潜艇遥感图像样本为预先采集的多张多方向和多尺度的潜艇图像集合；所述多方向潜艇图像，具体为水平方向不同角度拍摄、垂直方向不同角度拍摄的潜艇图像；所述多尺度潜艇图像，具体为被拍摄的潜艇，实际尺寸大小不同。