摘要
本发明公开了一种基于异构任务模型知识蒸馏的旋转小目标检测方法及系统,具体为:首先选择一种旋转目标检测算法作为基准旋转目标检测器,构造轻量级的特征图嵌入超分辨率模块嵌入基准旋转目标检测器的骨干网络与颈网络之间,得到超分辨率旋转目标检测器;然后构造特征编码模块、特征解码模块和特征上采样模块,将它们组合作为平行超分辨率网络分支;再利用特征仿射层和特征对齐损失对齐两类异构模型的输出特征,采用端到端联合训练同时优化嵌入特征图嵌入超分辨率模块的旋转目标检测器与平行超分辨率网络分支,实现异构任务模型间的知识蒸馏;最后使用训练好的超分辨率旋转目标检测器执行检测任务。本发明提高了旋转小目标的检测精度和检测速度。
技术关键词
检测器
输出特征
异构
超分辨率网络
蒸馏
分支
编码模块
解码模块
上采样
网络结构
基准
图像重建
梯度下降算法
超参数
压缩特征
抑制算法