[发明专利]一种目标识别方法、装置、电子设备和计算机存储介质有效
申请号: | 201910122654.4 | 申请日: | 2019-02-18 |
公开(公告)号: | CN109886185B | 公开(公告)日: | 2023-08-04 |
发明(设计)人: | 陈大鹏;赵瑞;李岁缠 | 申请(专利权)人: | 深圳市商汤科技有限公司 |
主分类号: | G06V40/10 | 分类号: | G06V40/10;G06V10/774;G06V10/82;G06N3/0442 |
代理公司: | 北京派特恩知识产权代理有限公司 11270 | 代理人: | 崔晓岚;张颖玲 |
地址: | 518054 广东省深圳市*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 目标 识别 方法 装置 电子设备 计算机 存储 介质 | ||
1.一种目标识别方法,其特征在于,所述方法包括:
构造训练数据集,包括:
从原始数据集中选取出目标数据集,所述目标数据集包括至少一个目标数据;针对所述目标数据集中的每一个目标数据,选取出所述每一个目标数据对应的参考数据集,所述参考数据集包括:与所述目标数据的标识信息相同的参考数据、以及与所述目标数据的标识信息不同的参考数据;所述参考数据和所述目标数据均为行人图片数据;所述目标数据的标识信息为所述目标数据对应的目标的身份标识;
根据所述目标数据和所述至少一个参考数据的相似程度,训练目标识别模型,得到训练完成的目标识别模型;
获取目标识别的初始结果序列,根据所述训练完成的目标识别模型,对所述初始结果序列进行重新排序,得到目标识别的重新排序结果。
2.根据权利要求1所述的方法,其特征在于,所述目标识别模型为双向的长短时记忆网络Bi-LSTM模型;所述Bi-LSTM模型的输入为所述目标数据和所述参考数据集的各个参考数据;所述Bi-LSTM模型的输出为所述目标数据和所述参考数据集各个参考数据的相似程度。
3.根据权利要求1或2所述的方法,其特征在于,所述根据所述训练完成的目标识别模型,对所述初始结果序列进行重新排序,包括:
根据所述训练完成的目标识别模型,确定所述目标数据与所述初始结果序列中前K个数据的相似程度,根据所述目标数据与所述初始结果序列中前K个数据的相似程度,对所述初始结果序列中前K个数据进行重新排序,K为大于1的整数。
4.根据权利要求1或2所述的方法,其特征在于,所述参考数据和所述目标数据均为图片数据。
5.一种目标识别装置,其特征在于,所述装置包括:构造模块、训练模块和重排序模块,其中,
构造模块,用于从原始数据集中选取出目标数据集,所述目标数据集包括至少一个目标数据;针对所述目标数据集中的每一个目标数据,选取出所述每一个目标数据对应的参考数据集,所述参考数据集包括:与所述目标数据的标识信息相同的参考数据、以及与所述目标数据的标识信息不同的参考数据;所述参考数据和所述目标数据均为行人图片数据;所述目标数据的标识信息为所述目标数据对应的目标的身份标识;
训练模块,根据所述目标数据和所述至少一个参考数据的相似程度,训练目标识别模型,得到训练完成的目标识别模型;
重排序模块,用于获取目标识别的初始结果序列,根据所述训练完成的目标识别模型,对所述初始结果序列进行重新排序,得到目标识别的重新排序结果。
6.根据权利要求5所述的装置,其特征在于,所述目标识别模型为双向的长短时记忆网络Bi-LSTM模型;所述Bi-LSTM模型的输入为所述目标数据和所述参考数据集的各个参考数据;所述Bi-LSTM模型的输出为所述目标数据和所述参考数据集各个参考数据的相似程度。
7.根据权利要求5或6所述的装置,其特征在于,所述重排序模块,具体用于根据所述训练完成的目标识别模型,确定所述目标数据与所述初始结果序列中前K个数据的相似程度,根据所述目标数据与所述初始结果序列中前K个数据的相似程度,对所述初始结果序列中前K个数据进行重新排序,K为大于1的整数。
8.根据权利要求5或6所述的装置,其特征在于,所述参考数据和所述目标数据均为图片数据。
9.一种电子设备,其特征在于,包括处理器和用于存储能够在处理器上运行的计算机程序的存储器;其中,
所述处理器用于运行所述计算机程序时,执行权利要求1至4任一项所述的方法。
10.一种计算机存储介质,其上存储有计算机程序,其特征在于,该计算机程序被处理器执行时实现权利要求1至4任一项所述的方法。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于深圳市商汤科技有限公司,未经深圳市商汤科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910122654.4/1.html,转载请声明来源钻瓜专利网。