一种基于知识蒸馏的轻量化SAR图像目标检测方法

    公开(公告)号:CN116935213A

    公开(公告)日:2023-10-24

    申请号:CN202310761441.2

    申请日:2023-06-26

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

    一种结合定位质量估计的开集SAR目标检测方法

    公开(公告)号:CN118196620A

    公开(公告)日:2024-06-14

    申请号:CN202410276443.7

    申请日:2024-03-12

    摘要: 本发明公开了一种结合定位质量估计的开集SAR目标检测方法,包括:获取SAR数据集并进行扩充,得到训练数据集;基于单阶段目标检测模型构建开集SAR目标检测器,在现有分类头中增加一路未知类判别分支,用以判别目标属于已知类还是未知类;同时,在现有回归头中增加一路定位质量估计分支,用于进行与类别无关的目标定位质量估计;引入未知类概率学习损失和定位质量估计损失对开集SAR目标检测器进行训练,以便于利用训练好的开集SAR目标检测器进行目标检测。该方法不仅能对已知类别进行检测识别,还能检测未知类目标,并将其判为未知类,减少了未知类目标和已知类目标的混淆,提升了SAR目标检测器在开放场景下的性能。