关键词:
遥感
多目标跟踪
小目标检测
注意力机制
神经网络
轨迹关联
摘要:
遥感卫星的多目标跟踪任务面监目标弱小,场景多样等挑战.为此,提出了一种高分辨率遥感卫星视频的多目标跟踪算法.在检测阶段,构建小目标检测器,首先在主干网络中通过Transformer捕获全局的上下文信息,然后利用注意力机制增强目标特征,最后添加了一个预测小目标的分支;在轨迹关联阶段,将检测出的小目标与已有轨迹匹配,采用关注低置信度检测的关联算法.本文选取高分辨率遥感卫星视频进行测试,测验结果表明本文提出的方法在遥感卫星视频中的多目标跟踪数据集上的MOTA指标达到63.1%,相较于基准(baseline)模型提升13.5%,能够显著提升遥感卫星视频中多目标跟踪的性能.