特征图知识蒸馏引导的轻量化任意方向SAR舰船目标检测器

基于有向边界框的合成孔径雷达(SAR)舰船目标检测器能输出精准的边界框，但仍存在模型计算复杂度高、推理速度慢、存储消耗大等问题，导致其难以在星载平台上部署。基于此该文提出了结合特征图和检测头分支知识蒸馏的无锚框轻量化旋转检测方法。首先，结合目标的长宽比和方向角信息提出改进高斯核，使生成的热度图能更好地刻画目标形状。然后在检测器预测头部引入前景区域增强分支，使网络更关注前景特征且抑制背景杂波的干扰。在训练轻量化网络时，将像素点间的相似度构建为热度图蒸馏知识。为解决特征蒸馏中正负样本不平衡问题，将前景注意力区域作为掩模引导网络蒸馏与目标相关的特征。另外，该文提出全局语义模块对像素进行上下文信息建模，能够结合背景知识加强目标精确表征。基于HRSID数据集的实验结果表明所提方法在模型参数仅有9.07 M的轻量化条件下，mAP能达到80.71%，且检测帧率满足实时应用需求。