Invention Publication
- Patent Title: 一种基于知识蒸馏的轻量化SAR图像目标检测方法
-
Application No.: CN202310761441.2Application Date: 2023-06-26
-
Publication No.: CN116935213APublication Date: 2023-10-24
- Inventor: 杜兰 , 麻弘扬 , 郭昱辰 , 周宇
- Applicant: 西安电子科技大学
- Applicant Address: 陕西省西安市太白南路2号
- Assignee: 西安电子科技大学
- Current Assignee: 西安电子科技大学
- Current Assignee Address: 陕西省西安市太白南路2号
- Agency: 西安嘉思特知识产权代理事务所
- Agent 王萌
- Main IPC: G06V20/10
- IPC: G06V20/10 ; G06V10/764 ; G06V10/82 ; G06N3/042 ; G06N3/045 ; G06N3/0495 ; G06N3/08

Abstract:
本发明公开了一种基于知识蒸馏的轻量化SAR图像的目标检测方法,包括:获取待检测的SAR图像;将所述待检测的SAR图像输入至训练好的轻量化的学生网络,得到目标检测结果;其中,以YOLOX‑S模型为基线构建所述轻量化的学生网络;轻量化的学生网络利用训练好的教师网络进行训练,且轻量化的学生网络的损失函数包括非蒸馏损失与蒸馏损失,以使训练好的轻量化的学生网络结合顶层知识蒸馏与隐层知识蒸馏的知识蒸馏方法进行目标检测,得到目标检测结果。本发明能够充分提取教师网络中有利于轻量化的学生网络学习的知识,从而更好地提高轻量化的性能,使得轻量化模型的检测精度可与复杂大模型相比较,实现对地检测的精度要求。
Information query