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