文献类型：期刊文献

中文题名：引入注意力机制的JDE多目标跟踪方法

英文题名：JDE Multi-Object Tracking Method with Attention Mechanism

作者：晏康[1];曾凤彩[1];何宁[1];贺宇哲[2];张人[2]

第一作者：晏康

机构：[1]北京联合大学智慧城市学院,北京100101;[2]北京联合大学北京市信息服务工程重点实验室,北京100101

第一机构：北京联合大学智慧城市学院

年份：2022

卷号：58

期号：21

起止页码：189-196

中文期刊名：计算机工程与应用

外文期刊名：Computer Engineering and Applications

收录：CSTPCD;;北大核心:【北大核心2020】；CSCD:【CSCD_E2021_2022】；

基金：国家自然科学基金(61872042,61572077);北京市教委科技重点项目(KZ201911417048);北京联合大学人才强校优选计划(BPHR2020AZ01,BPHR2020EZ01);北京联合大学科研项目(ZK50202001);国家重点研发计划(2018AAA0100804)。

语种：中文

中文关键词：多目标跟踪;注意力机制;多尺度融合;特征增强;JDE算法

外文关键词：multi-object tracking;attention mechanism;multi-scale fusion;feature enhancement;JDE algorithm

摘要：多目标跟踪是计算机视觉领域的一个重要研究内容。JDE(joint detection and embedding)多目标跟踪算法推理速度和精度较高,但是当目标重叠或尺度较小时,该算法的跟踪效果较差。针对以上问题,提出了Attention-JDE,该模型结合了注意力机制、多尺度融合等思想,利用特征金字塔(feature pyramid)和空间金字塔池化(spatial pyramid pooling)提升模型对于小尺度目标的检测和跟踪能力,结合空间域注意力机制和通道域注意力机制改进模型在目标发生重叠时的跟踪效果。此外,还引入了Mish激活函数有效地降低跟踪时的ID切换次数。在MOT16数据集进行验证,结果表明,与原JDE方法以及其他主流方法相比,Attention-JDE具有更高的跟踪精度(MOTA),同时速度能够达到19.5 FPS,实时性较高。
Multi-object tracking is an important research content in the field of computer vision.JDE(joint detection and embedding)multi-object tracking algorithm has high inference speed and accuracy,but when the object overlap or scale is small,the tracking performance of the algorithm is bad.To solve the above problems,this paper proposes AttentionJDE,which combines attention mechanism,multi-scale fusion and other ideas,uses feature pyramid and spatial pyramid pooling to improve the detection and tracking ability of the model for small-scale objects.The spatial attention mechanism and channel attention mechanism are combined to improve the tracking performance when the objects overlap.In addition,this paper also introduces the Mish activation function to effectively reduce the number of ID switching during tracking.Compared with the original JDE method and other SOTA methods,Attention-JDE has higher tracking accuracy(MOTA),inference speed can reach to 19.5 FPS which is high real-time performance.