一种基于深度学习的遥感图像水面潜艇识别方法

摘要

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