北京京航计算通讯研究所王东豪获国家专利权
买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
龙图腾网获悉北京京航计算通讯研究所申请的专利一种基于知识蒸馏的目标检测模型的训练方法获国家发明授权专利权,本发明授权专利权由国家知识产权局授予,授权公告号为:CN119295739B 。
龙图腾网通过国家知识产权局官网在2026-01-16发布的发明授权授权公告中获悉:该发明授权的专利申请号/专利号为:202411452110.1,技术领域涉及:G06V10/25;该发明授权一种基于知识蒸馏的目标检测模型的训练方法是由王东豪;李姝;高晓琼;唱明旭;杨波设计研发完成,并于2024-10-17向国家知识产权局提交的专利申请。
本一种基于知识蒸馏的目标检测模型的训练方法在说明书摘要公布了:本发明涉及一种基于知识蒸馏的目标检测模型的训练方法,属于目标检测领域。包括:构建训练数据集及目标检测模型,基于训练数据集对目标检测模型进行训练输出推理图;计算第一损失函数得到训练后的目标检测模型,将其作为教师模型;将教师模型的参数进行简化得到学生模型;基于训练数据集得到第二训练数据集,基于第二训练数据集对教师模型及学生模型进行训练分别输出预测标签;计算第二损失函数得到训练好的学生模型作为训练好的目标检测模型。该方法采用差分变异算法得到最优的蒸馏温度及损失系数,通过知识蒸馏训练学生模型得到训练好的目标检测模型,解决了现有目标检测模型在少量样本训练的情况下,导致检测精度低的问题。
本发明授权一种基于知识蒸馏的目标检测模型的训练方法在权利要求书中公布了:1.一种基于知识蒸馏的目标检测模型的训练方法,其特征在于,所述方法包括以下步骤: 对公开数据集中的图像进行预处理,并标注后构建训练数据集;构建目标检测模型,将训练数据集中的每个样本输入所述目标检测模型进行训练,输出对应的推理图;所述推理图包括预测框; 基于YOLOv5s模型构建初始目标检测模型,所述初始目标检测模型包括输入端、骨干网络、颈部网络和检测端; 通过样本的标注框与对应的推理图的预测框之间的误差计算第一损失函数,当第一损失函数达到阈值时得到训练后的目标检测模型,将其作为教师模型;将所述教师模型的骨干网络的卷积通道数减半得到学生模型; 所述第一损失函数的表达式如下所示: Ltotal1=Lo2m+Lo2o Lo2m=λobjBCEobj+λclsBCEcls+λboxEIoU Lo2o=δ11BCEcls+δ2IoU, 其中,Ltotal1表示第一损失函数,Lo2m表示One-to-manyDetect网络结构的损失函数,Lo2o表示One-to-oneDetect网络结构的损失函数,BCEobj为检测目标置信度损失,BCEcls为检测目标类别损失,EIoU为One-to-manyDetect网络结构的预测框损失,IoU为One-to-oneDetect网络结构的预测框损失,λobj、λcls、λbox、δ1、δ2为权重系数; 基于所述训练数据集得到第二训练数据集,将所述第二训练数据集中的每个样本分别输入所述教师模型以及学生模型进行训练,所述教师模型以及学生模型分别输出对应的预测标签;通过样本的真实标签与学生模型对应的预测标签之间的误差以及教师模型对应的预测标签与学生模型对应的预测标签之间的误差计算第二损失函数,当第二损失函数达到阈值时得到训练好的学生模型,将训练好的学生模型作为训练好的目标检测模型。
如需购买、转让、实施、许可或投资类似专利技术,可联系本专利的申请人或专利权人北京京航计算通讯研究所,其通讯地址为:100074 北京市丰台区云岗北区西里1号院;或者联系龙图腾网官方客服,联系龙图腾网可拨打电话0551-65771310或微信搜索“龙图腾网”。
以上内容由龙图腾AI智能生成。
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。

皖公网安备 34010402703815号
请提出您的宝贵建议,有机会获取IP积分或其他奖励