[发明专利]网络训练、识别方法及装置、电子设备和存储介质在审
申请号: | 201911012578.8 | 申请日: | 2019-10-23 |
公开(公告)号: | CN110765943A | 公开(公告)日: | 2020-02-07 |
发明(设计)人: | 陈裕汀;朱烽;赵瑞 | 申请(专利权)人: | 深圳市商汤科技有限公司 |
主分类号: | G06K9/00 | 分类号: | G06K9/00 |
代理公司: | 11277 北京林达刘知识产权代理事务所(普通合伙) | 代理人: | 刘新宇 |
地址: | 518054 广东省深圳市*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 目标样本 标签 损失函数 特征表示 特征提取 网络训练 样本图片 负样本 正样本 图片 存储介质 电子设备 计算资源 网络 | ||
1.一种网络训练方法,其特征在于,包括:
采用待训练的特征提取网络,提取目标样本图片的特征;
获取所述目标样本图片对应的正特征和负特征,其中,所述正特征表示正样本的特征,所述负特征表示负样本的特征,所述正样本表示对应标签与所述目标样本图片的标签相同的其他样本图片,所述负样本表示对应标签与所述目标样本图片的标签不同的样本图片;
根据所述目标样本图片的特征、所述目标样本图片对应的正特征和负特征,确定损失函数值;
基于所述损失函数值更新所述特征提取网络的参数。
2.根据权利要求1所述的方法,其特征在于,获取所述目标样本图片对应的正特征和负特征,包括:
将第一特征集合中对应标签与所述目标样本图片的标签相同的标签特征,确定为所述目标样本图片对应的正特征;
将第二特征集合中对应标签与所述目标样本图片的标签不同的特征,确定为所述目标样本图片对应的负特征。
3.根据权利要求2所述的方法,其特征在于,样本集合中包含针对每个标签的样本图片,每个标签对应至少一个样本图片,所述方法还包括:
采用所述待训练的特征提取网络,提取所述样本集合中的每个所述样本图片的特征;
根据所提取的特征,确定每个标签对应的标签特征;
基于所有标签对应的标签特征,构建所述第一特征集合。
4.根据权利要求3所述的方法,其特征在于,所述方法还包括:
从所提取的特征中选取指定数量的特征,构建所述第二特征集合。
5.根据权利要求2至4中任意一项所述的方法,其特征在于,所述方法还包括:
基于所述目标样本图片的特征以及所述目标样本图片对应的正特征,更新所述第一特征集合。
6.一种识别方法,其特征在于,包括:
采用特征提取网络,提取测试图片的特征;
根据所述测试图片的特征确定所述测试图片对应的标签;
其中,所述特征提取网络是基于目标样本图片的特征、目标样本图片对应的正特征和目标样本图片对应的负特征训练完成的;
所述正特征表示正样本的特征,所述负特征表示负样本的特征,所述正样本表示对应标签与所述目标样本图片的标签相同的其他样本图片,所述负样本表示对应标签与所述目标样本图片的标签不同的样本图片。
7.一种网络训练装置,其特征在于,包括:
第一提取模块,用于采用待训练的特征提取网络,提取目标样本图片的特征;
获取模块,用于获取所述目标样本图片对应的正特征和负特征,其中,所述正特征表示正样本的特征,所述负特征表示负样本的特征,所述正样本表示对应标签与所述目标样本图片的标签相同的其他样本图片,所述负样本表示对应标签与所述目标样本图片的标签不同的样本图片;
第一确定模块,用于根据所述目标样本图片的特征、所述目标样本图片对应的正特征和负特征,确定损失函数值;
第一更新模块,用于基于所述损失函数值更新所述特征提取网络的参数。
8.一种识别装置,其特征在于,包括:
提取模块,用于采用特征提取网络,提取测试图片的特征;
测试模块,用于根据所述测试图片的特征确定所述测试图片对应的标签;
其中,所述特征提取网络是基于目标样本图片的特征、目标样本图片对应的正特征和目标样本图片对应的负特征训练完成的;
所述正特征表示正样本的特征,所述负特征表示负样本的特征,所述正样本表示对应标签与所述目标样本图片的标签相同的其他样本图片,所述负样本表示对应标签与所述目标样本图片的标签不同的样本图片。
9.一种电子设备,其特征在于,包括:
处理器;
用于存储处理器可执行指令的存储器;
其中,所述处理器被配置为调用所述存储器存储的指令,以执行权利要求1至5中任意一项或者权利要求6所述的方法。
10.一种计算机可读存储介质,其上存储有计算机程序指令,其特征在于,所述计算机程序指令被处理器执行时实现权利要求1至5中任意一项或者权利要求6所述的方法。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于深圳市商汤科技有限公司,未经深圳市商汤科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201911012578.8/1.html,转载请声明来源钻瓜专利网。