← 返回列表
目标检测训练方法及电子设备、计算机可读存储介质
申请人信息
- 申请人:深圳华付技术股份有限公司
- 申请人地址:518000 广东省深圳市前海深港合作区前湾一路1号A栋201室(入驻深圳市前海商务秘书有限公司)
- 发明人: 深圳华付技术股份有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 目标检测训练方法及电子设备、计算机可读存储介质 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410060088.X |
| 申请日 | 2024/1/16 |
| 公告号 | CN117576381A |
| 公开日 | 2024/2/20 |
| IPC主分类号 | G06V10/25 |
| 权利人 | 深圳华付技术股份有限公司 |
| 发明人 | 张欢; 娄德成; 刘魏魏; 陈兴委; 赵士红 |
| 地址 | 广东省深圳市前海深港合作区前湾一路1号A栋201室(入驻深圳市前海商务秘书有限公司) |
摘要文本
来自： 本申请公开了一种目标检测训练方法及电子设备、计算机可读存储介质。构建教师模型和学生模型，完成对教师模型的训练，以得到训练好的教师模型；计算教师模型和学生模型之间的特征蒸馏损失、响应蒸馏损失及交叉蒸馏损失，以及计算学生模型的强监督损失；并构建学生模型的总损失函数，基于总损失函数训练学生模型，优化学生模型的参数，以得到最终的学生模型。本申请通过提取多个中间层特征进行特征蒸馏让学生模型更加全面地学习教师模型的特征表示；并通过交叉蒸馏使学生模型学习教师模型的学习能力。以此提高学生模型学习教师模型的能力，从而提升学生模型检测精确度，使得学生模型能够更好地应用在设备上。
专利主权项内容
1.一种目标检测训练方法，其特征在于，包括：构建教师模型和学生模型，其中所述学生模型是通过将所述教师模型的通道数和网络层数进行缩减之后得到的；完成对所述教师模型的训练，以得到训练好的所述教师模型；计算所述教师模型和所述学生模型之间的特征蒸馏损失、响应蒸馏损失及交叉蒸馏损失，以及计算所述学生模型的强监督损失；基于所述特征蒸馏损失、所述响应蒸馏损失、所述交叉蒸馏损失和所述强监督损失构建所述学生模型的总损失函数；基于所述总损失函数训练所述学生模型，优化所述学生模型的参数，以得到最终的所述学生模型。