一种基于特征的红外小目标检测知识蒸馏方法

    公开(公告)号:CN117671289A

    公开(公告)日:2024-03-08

    申请号:CN202311595168.7

    申请日:2023-11-27

    Abstract: 本发明公开了一种基于特征的红外小目标检测知识蒸馏方法,涉及图像处理和计算机视觉技术领域,包括:首先提取教师模型和学生模型的特征图,得到教师特征图和学生特征图;再对教师特征图和学生特征图进行空间‑通道维度解耦,解耦后获得空间‑通道两个维度特征图;然后通过空间‑通道注意力模块强调教师模型和学生模型的细节差异,并指导训练学生模型;最后获得训练好的学生模型;本发明,不仅具备出色的鲁棒性,并且能灵活应用于多种检测框架,不增加计算复杂度的前提下显著提升了检测性能,与其他知识蒸馏方法相比,所训练的学生检测器在平均精度上有了明显的提高。

Patent Agency Ranking