有效

用于目标跟踪的知识蒸馏方法及系统

罗来龙、潘永淇、朱承、朱先强、张千桢、张胜
中国人民解放军国防科技大学

摘要

本发明提供了一种用于目标跟踪的知识蒸馏方法及系统,涉及目标跟踪技术领域。通过基于掩膜的特征蒸馏,从教师模型的特征融合层中提取空间注意力和通道注意力,生成注意力掩膜,并将注意力掩膜与学生模型的特征图相乘;基于上下文自适应学习的方式,通过多层感知器自适应地集成教师模型各层的空间和特征知识以使学生模型能够根据网络的不同深度自适应地调整学习重点;通过相关距离监督进行蒸馏,采用正仿射变换将教师模型和学生模型的回归框预测投影到概率分布空间中,并通过评估概率分布的相关距离来监督教师模型和学生模型之间的能力差距。实现了在目标跟踪的过程中提升通用性,并且优于现有基线的性能的技术效果。