基于结构化实例图的知识蒸馏方法、装置、设备和介质
基本信息
申请号 | CN202110551061.7 | 申请日 | - |
公开(公告)号 | CN113255915A | 公开(公告)日 | 2021-08-13 |
申请公布号 | CN113255915A | 申请公布日 | 2021-08-13 |
分类号 | G06N5/02(2006.01)I;G06K9/62(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 陈亦新;陈鹏光;贾佳亚;沈小勇;吕江波 | 申请(专利权)人 | 深圳思谋信息科技有限公司 |
代理机构 | 广州华进联合专利商标代理有限公司 | 代理人 | 潘思延 |
地址 | 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼 | ||
法律状态 | - |
摘要
摘要 | 本申请涉及一种基于结构化实例图的知识蒸馏方法、装置、计算机设备和存储介质。本申请能够使得教师网络中的前景特征与背景特征都能够被有效地迁移到学生网络中,进一步提高了目标检测的准确度。该方法包括:获取训练图像,分别将训练图像输入至教师网络学生网络,得到第一特征图和第二特征图;再分别将第一特征图和第二特征图输入区域候选网络中,得到包含边界框的第一待检测图和第二待检测图;基于各边界框中的对象实例进行编码,得到第一结构图和第二结构图;基于第一结构图和第二结构图,得到蒸馏损失部分;根据第二待检测图的检测结果与真实标签之间的距离得到基础损失部分;基于蒸馏损失部分和基础损失部分,对学生网络进行训练。 |
