[发明专利]神经网络训练及图像分类方法、装置、存储介质、设备在审

专利信息
申请号: 202010833576.1 申请日: 2020-08-18
公开(公告)号: CN111967597A 公开(公告)日: 2020-11-20
发明(设计)人: 李潇婕;王飞;钱晨 申请(专利权)人: 上海商汤临港智能科技有限公司
主分类号: G06N3/08 分类号: G06N3/08;G06K9/62
代理公司: 北京博思佳知识产权代理有限公司 11415 代理人: 靳玫
地址: 200232 上海市浦东新区(上海)自由贸易试*** 国省代码: 上海;31
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 神经网络 训练 图像 分类 方法 装置 存储 介质 设备
【权利要求书】:

1.一种神经网络训练方法,其特征在于,包括:

将样本图像分别输入教师网络和学生网络,得到所述教师网络输出的第一特征图,以及所述学生网络输出的第二特征图;

基于所述第一特征图包括的特征信息,确定第一关联特征信息,以及基于所述第二特征图包括的特征信息,确定第二关联特征信息;

基于所述第二关联特征信息相对于所述第一关联特征信息的差异,对所述学生网络进行监督训练。

2.根据权利要求1所述的方法,其特征在于,所述方法还包括:

基于所述第一特征图,确定目标区域;

基于所述目标区域以及所述第一特征图,确定第一目标特征图,以及基于所述目标区域以及所述第二特征图,确定第二目标特征图;

所述基于所述第一特征图包括的特征信息,确定第一关联特征信息,以及基于所述第二特征图包括的特征信息,确定第二关联特征信息,包括:

基于所述第一目标特征图包括的特征信息,确定所述第一关联特征信息,以及基于所述第二目标特征图包括的特征信息,确定所述第二关联特征信息。

3.根据权利要求2所述的方法,其特征在于,所述样本图像的所述目标区域包括目标对象,和/或,所述样本图像中除所述目标区域以外的区域包括如下至少一项:背景部分、所述目标对象被遮挡的部分。

4.根据权利要求2或3所述的方法,其特征在于,所述基于所述第一特征图,确定目标区域,包括:

将所述第一特征图输入预先训练好的第一神经网络,得到所述第一神经网络输出的所述样本图像包括的多个区域中每个区域对应的像素归一化值;

将所述像素归一化值大于或等于预设值的区域作为所述目标区域。

5.根据权利要求4所述的方法,其特征在于,所述基于所述目标区域以及所述第一特征图,确定第一目标特征图,以及基于所述目标区域以及所述第二特征图,确定第二目标特征图,包括:

将所述多个区域中每个区域对应的像素归一化值与所述第一特征图包括的特征信息进行点乘,得到第一注意力特征图,以及将所述多个区域中每个区域对应的像素归一化值与所述第二特征图包括的特征信息进行点乘,得到第二注意力特征图;

基于所述第一注意力特征图上的所述目标区域,得到所述第一目标特征图,以及基于所述第二注意力特征图上的所述目标区域,得到所述第二目标特征图。

6.根据权利要求4或5所述的方法,其特征在于,所述第一神经网络包括多个第一网络层和第一输出层,第二神经网络包括多个第二网络层和第二输出层,其中,所述多个第二网络层的网络结构与所述多个第一网络层的网络结构相同;

所述方法还包括:

将所述第一特征图作为所述第二神经网络的输入,得到所述第二神经网络输出的分类结果;

基于所述第二神经网络输出的分类结果相对于所述样本图像中标注的分类结果的差异,对所述第二神经网络进行监督训练;

将训练完成的所述第二神经网络包括的所述多个第二网络层的网络参数,作为所述第一神经网络包括的所述多个第一网络层的网络参数。

7.根据权利要求1-6任一项所述的方法,其特征在于,所述基于所述第一特征图包括的特征信息,确定第一关联特征信息,包括:

基于所述第一特征图上不同区域的特征信息之间的相似度,确定所述第一关联特征信息;

所述基于所述第二特征图包括的特征信息,确定第二联特征信息,包括:

基于所述第二特征图上不同区域的特征信息之间的相似度,确定所述第二关联特征信息。

下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海商汤临港智能科技有限公司,未经上海商汤临港智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服

本文链接:http://www.vipzhuanli.com/pat/books/202010833576.1/1.html,转载请声明来源钻瓜专利网。

×

专利文献下载

说明:

1、专利原文基于中国国家知识产权局专利说明书;

2、支持发明专利 、实用新型专利、外观设计专利(升级中);

3、专利数据每周两次同步更新,支持Adobe PDF格式;

4、内容包括专利技术的结构示意图流程工艺图技术构造图

5、已全新升级为极速版,下载速度显著提升!欢迎使用!

请您登陆后,进行下载,点击【登陆】 【注册】

关于我们 寻求报道 投稿须知 广告合作 版权声明 网站地图 友情链接 企业标识 联系我们

钻瓜专利网在线咨询

周一至周五 9:00-18:00

咨询在线客服咨询在线客服
tel code back_top