水上目标识别方法、装置及电子设备和存储介质与流程

文档序号:23340101发布日期:2020-12-18 16:35阅读:来源:国知局

技术特征:

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

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

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

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

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

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

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

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

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

对该候选区域进行特征提取,得到区域特征;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

9.一种电子设备,包括:

存储器,用于至少存储一组指令集;

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

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


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

技术研发人员:姜艳川;姚力波;孙炜玮;计祥奇;郭亚静;张学宇;郑文旭;于梦馨
受保护的技术使用者:北京瞰天科技有限公司
技术研发日:2020.09.25
技术公布日:2020.12.18
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1