融合注意力机制的单步双分支多目标跟踪算法

Single-step dual-branch multi-object tracking algorithm fused with attention mechanism

为提高多目标跟踪的实时性、解决形变和遮挡后的身份切换问题,提出一种融合改进的通道注意力机制的单步双分支多目标跟踪算法。以深度聚合网络为框架,通过改进通道注意力机制,充分利用多层次特征并考虑特征通道依赖关系;联合训练目标检测分支与重识别分支,增强检测鲁棒性和模型抗遮挡能力;利用外观、运动以及IOU信息进行数据关联。实验结果表明,使用该算法有效缓解了多目标跟踪遇到的目标形变和遮挡问题带来的影响,性能和效率得到大幅提升。

To improve the real-time performance of multi-object tracking and solve the problem of identity switching after defor-mation and occlusion,a single-step dual-branch multi-object tracking algorithm incorporating an improved channel attention mechanism was proposed.Taking the deep aggregation network as the framework,by improving the channel attention mechanism,multi-level features were utilized and the dependency of feature channels was considered.The target detection branch and re-identification branch were jointly trained to enhance the detection robustness and the model’s anti-occlusion ability.The appearance,movement,and IOU information were used for data association.Experimental results show that the effects of target deformation and occlusion problems encountered in multi-object tracking are effectively alleviated,and the performance and efficiency are greatly improved.