基于知识蒸馏的目标检测模型训练方法

发明专利有效专利

申请号：
CN202111179182.X
IPC分类号：G06K9/00;G06K9/32;G06K9/62;G06N20/00
申请日期：
2021-10-11
申请人：
北京文安智能技术股份有限公司

基础信息

权利要求

说明书

PDF全文

法律信息

引证文献

著录项信息

专利名称	基于知识蒸馏的目标检测模型训练方法
申请号	CN202111179182.X	申请日期	2021-10-11
法律状态	公开	申报国家	暂无
公开/公告日	2021-11-05	公开/公告号	CN113610069A
优先权	暂无	优先权号	暂无
主分类号	G06K9/00 ? IPC结构图谱： G 物理 G0 仪器 G06 计算；推算；计数 G06K 数据识别；数据表示；记录载体；记录载体的处理（印刷本身入B41J） G06K9/00 用于阅读或识别印刷或书写字符或者用于识别图形，例如，指纹的方法或装置（用于图表阅读或者将诸如力或现状态的机械参量的图形转换为电信号的方法或装置入G06K 11/00；语音识别入G10L 15/00）〔1，7〕	IPC分类号	G;0;6;K;9;/;0;0;;;G;0;6;K;9;/;3;2;;;G;0;6;K;9;/;6;2;;;G;0;6;N;2;0;/;0;0查看分类表>
申请人	北京文安智能技术股份有限公司	申请人地址	北京市海淀区丰豪东路9号院2号楼5单元801 变更专利地址、主体等相关变化，请及时变更，防止失效
权利人	北京文安智能技术股份有限公司	当前权利人	北京文安智能技术股份有限公司
发明人	张志嵩;曹松;任必为;宋君;陶海
代理机构	暂无	代理人	暂无

摘要

本发明提供了一种基于知识蒸馏的目标检测模型训练方法，包括：利用训练样本图像集训练目标检测教师模型，训练样本图像具有：第一标签：目标检测框中心点像素位置硬标签概率矩阵；第二标签：目标检测框的宽和高；第三标签：目标检测框中心点像素位置偏移量；目标检测教师模型的预测输出结果包括：目标检测框中心点像素位置概率热力图、目标检测框的宽和高、目标检测框中心点像素位置偏移量；以知识蒸馏的方式改进目标检测学生模型的损失函数后，训练生成目标检测学生模型。本发明的解决了利用现有的知识蒸馏方法训练获取的目标检测模型无法同时保证网络结构简单而满足终端设备使用需求，以及目标检测模型的识别率优良以确保模型检测精度的问题。

序号	公开(公告)号	公开(公告)日	申请日	专利名称	申请人
该专利没有引用任何外部专利数据！

序号	公开(公告)号	公开(公告)日	申请日	专利名称	申请人
该专利没有被任何外部专利所引用！

我浏览过的专利

专利服务由北京酷爱智慧知识产权代理公司提供