一种基于知识蒸馏的目标检测方法及目标检测终端.pdfVIP

  • 10
  • 0
  • 约1.47万字
  • 约 15页
  • 2023-07-02 发布于四川
  • 举报

一种基于知识蒸馏的目标检测方法及目标检测终端.pdf

本发明属于深度学习技术领域,具体涉及一种基于知识蒸馏的目标检测方法及目标检测终端,所述方法包括以下步骤:获取图像数据;将图像数据输入训练好的目标检测模型;目标检测模型输出检测结果;其中,所述的目标检测模型经过以下步骤训练得到:构建初始的教师网络,对教师网络进行压缩得到学生网络;通过注意力加权的特征蒸馏方式利用训练好的教师网络对学生网络进行知识蒸馏,得到蒸馏后的学生网络;对蒸馏后的学生网络进行训练与测试,得到最终的目标检测模型。本发明通过对网络的压缩、蒸馏优化和训练得到目标检测模型,蒸馏优化能够在

(19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 113743514 A (43)申请公布日 2021.12.03 (21)申请号 202111048823.8 (22)申请日 2021.09.08 (71)申请人 庆阳瑞华能源有限公司

文档评论(0)

1亿VIP精品文档

相关文档