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

权利要求书

1.一种水上目标识别方法，其特征在于，包括：

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述目标识别模型通过如下方式训练得到：

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

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

3.根据权利要求1所述的方法，其特征在于，所述确定该候选区域内是否存在水上航行设备，包括：

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

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

4.根据权利要求1所述的方法，其特征在于，所述获取所述水上航行设备的剪影，包括：

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

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

5.根据权利要求1所述的方法，其特征在于，所述多角度至少包括：正前侧、正后侧、正左侧、正右侧、左前斜侧、右前斜侧、左后斜侧、右后斜侧。

6.一种水上目标识别装置，其特征在于，包括：

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

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

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

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

7.根据权利要求6所述的装置，其特征在于，还包括：

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

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

8.根据权利要求6所述的装置，其特征在于，所述设备确定模块包括：

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

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

9.一种电子设备，包括：

存储器，用于至少存储一组指令集；

处理器，用于调用并执行所述存储器中的所述指令集，通过执行所述指令集实现权利要求1-5中任意一项所述的水上目标识别方法的各个步骤。

10.一种可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时，实现如权利要求1-5中任一项所述的水上目标识别方法的各个步骤。