授权公布号:CN113610069B
基于知识蒸馏的目标检测模型训练方法
有效
申请
2021-10-11
申请公布
2021-11-05
授权
2022-02-08
预估到期
2041-10-11
| 申请号 | CN202111179182.X |
| 申请日 | 2021-10-11 |
| 申请公布号 | CN113610069A |
| 申请公布日 | 2021-11-05 |
| 授权公布号 | CN113610069B |
| 授权公告日 | 2022-02-08 |
| 分类号 | G06V10/25;G06V10/764;G06V10/774;G06V40/10;G06K9/62;G06N20/00 |
| 分类 | 计算;推算;计数; |
| 申请人名称 | 北京文安智能技术股份有限公司 |
| 申请人地址 | 北京市海淀区丰豪东路9号院2号楼5单元801 |
专利法律状态
2022-02-08
授权
状态信息
授权
2021-11-05
公布
状态信息
公布
摘要
本发明提供了一种基于知识蒸馏的目标检测模型训练方法,包括:利用训练样本图像集训练目标检测教师模型,训练样本图像具有:第一标签:目标检测框中心点像素位置硬标签概率矩阵;第二标签:目标检测框的宽和高;第三标签:目标检测框中心点像素位置偏移量;目标检测教师模型的预测输出结果包括:目标检测框中心点像素位置概率热力图、目标检测框的宽和高、目标检测框中心点像素位置偏移量;以知识蒸馏的方式改进目标检测学生模型的损失函数后,训练生成目标检测学生模型。本发明的解决了利用现有的知识蒸馏方法训练获取的目标检测模型无法同时保证网络结构简单而满足终端设备使用需求,以及目标检测模型的识别率优良以确保模型检测精度的问题。


