基于知识蒸馏的旋转目标检测轻量化方法及系统.pdfVIP

  • 3
  • 0
  • 约2.23万字
  • 约 23页
  • 2023-08-02 发布于四川
  • 举报

基于知识蒸馏的旋转目标检测轻量化方法及系统.pdf

本发明涉及一种基于知识蒸馏的旋转目标检测轻量化方法及系统,方法包括将学生模型和训练好的教师模型输出的检测目标的坐标编码分别解码为旋转坐标形式;将旋转坐标转换为二维高斯分布;计算学生模型与教师模型输出的检测目标的高斯分布表示的KL散度,以及计算学生模型输出的检测目标的高斯分布表示与标签真实框得到的检测目标的高斯分布表示间的KL散度;对KL散度进行归一化处理;根据归一化处理后的KL散度计算得到教师模型对学生模型进行知识蒸馏过程中的总体损失函数;使用蒸馏后的学生模型进行预测。本发明为轻量化网络的训练提

(19)国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 116524351 A (43)申请公布日 2023.08.01 (21)申请号 202310299030.6 G06N 3/096 (2023.01) (22)申请日 2023.03.

文档评论(0)

1亿VIP精品文档

相关文档