[发明专利]水上目标识别方法、装置及电子设备和存储介质

说明书

本申请实施例公开了一种水上目标识别方法、装置及电子设备和存储介质，包括：将目标图像输入预先训练好的区域候选网络模型，以在目标图像中确定候选区域；对应每一个候选区域，若该候选区域内存在水上航行设备，获取水上航行设备的剪影，水上航行设备的大小，以及水上航行设备距离目标图像的采集设备的距离；将剪影、水上航行设备的大小以及距离输入预先训练好的目标识别模型，得到水上航行设备的型号；其中，训练目标识别模型使用的训练数据通过对若干个型号的水上航行设备在各种模拟环境下的三维模型进行多角度多尺寸的投影得到，能够获取大量的训练样本，从而提高目标识别模型的识别精度。

技术领域

本申请涉及图像处理技术领域，更具体地说，涉及一种水上目标识别方法、装置及电子设备和存储介质。

背景技术

在港口巡逻或水上航行过程中，有发现和识别水上航行设备(比如，航母、两栖装备、巡洋舰、驱逐舰、护卫舰、登陆舰、扫雷舰、潜艇、快艇，以及相关支援保障船等)的需求。传统的水上航行设备的发现与识别主要依靠人的经验，存在受环境影响大，发现距离小等问题，不符合现代舰船航行和作战的需要。

目前，有方案提出采用神经网络模型进行水上目标识别，但训练神经网络模型需要大量样本，而目前用于水上目标识别的样本数量非常少，导致训练得到的神经网络模型的识别准确率较低。

发明内容

本申请的目的是提供一种水上目标识别方法、装置及电子设备和存储介质，包括如下技术方案：

一种水上目标识别方法，包括：

将目标图像输入预先训练好的区域候选网络模型，以在所述目标图像中确定候选区域；

对应每一个候选区域，确定该候选区域内是否存在水上航行设备；

若该候选区域内存在水上航行设备，获取所述水上航行设备的剪影，以及所述水上航行设备的大小，以及所述水上航行设备距离所述目标图像的采集设备的距离；

将所述剪影、所述水上航行设备的大小以及所述距离输入预先训练好的目标识别模型，得到所述水上航行设备的型号；其中，训练所述目标识别模型使用的训练数据集通过对若干个型号的水上航行设备在各种模拟环境下的三维模型进行多角度多尺寸的投影得到。

上述方法，优选的，所述目标识别模型通过如下方式训练得到：

将训练样本、所述训练样本中的水上航行设备的大小以及所述训练样本中的水上航行设备距离训练样本的采集设备的距离输入待训练的目标识别模型，得到识别结果，所述识别结果表征所述训练样本中的水上航行设备的型号；

根据所述识别结果与所述训练样本的标签的差异，对所述待训练的目标识别模型的参数进行更新；所述标签表征所述训练样本中的水上航行设备的真实型号。

上述方法，优选的，所述确定该候选区域内是否存在水上航行设备，包括：

对该候选区域进行特征提取，得到区域特征；

将所述区域特征输入预先训练好的二分类器，得到分类结果，所述分类结果表征所述候选区域内是否存在水上航行设备。

上述方法，优选的，所述获取所述水上航行设备的剪影，包括：

对该候选区域依次进行增强反差、边缘增强和轮廓提取，得到所述水上航行设备的轮廓；

将所述轮廓包围的区域确定为所述水上航行设备的剪影。

上述方法，优选的，所述多角度至少包括：正前侧、正后侧、正左侧、正右侧、左前斜侧、右前斜侧、左后斜侧、右后斜侧。

一种水上目标识别装置，包括：

候选区域确定模块，用于将目标图像输入预先训练好的区域候选网络模型，以在所目标图像中确定候选区域；