###
计算机系统应用英文版:2021,30(12):128-138
←前一篇   |   后一篇→
本文二维码信息
码上扫一扫!
基于注意力机制和卡尔曼滤波的多目标跟踪
(西南交通大学 电气工程学院, 成都 611756)
Multi-Target Tracking Using Attention Mechanism and Kalman Filter
(School of Electrical Engineering, Southwest Jiaotong University, Chengdu 611756, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 895次   下载 1950
Received:March 02, 2021    Revised:March 29, 2021
中文摘要: 为了解决目前多目标跟踪算法在行人遮挡后无法再次准确跟踪的问题, 提出了一种融入注意力机制和改进卡尔曼滤波的多目标跟踪算法, 选用联合检测和重识别框架提取特征, 同时完成目标检测和重识别任务. 设计了并行支路注意力机制, 包括空间注意力和通道注意力两个部分, 每个部分都采用并行支路的方式完成池化和卷积操作. 在跟踪阶段, 本文提出了速度先验卡尔曼滤波, 实现对行人运动状态更精确的预测. 采用CUHK-SYSU数据集对算法进行训练, 并在MOT16数据集上做算法的验证和测试. 本算法的多目标跟踪准确度(MOTA)达到了65.1%, 多目标跟踪精确度(MOTP)达到了78.8%, 识别F1值(IDF1)达到62.3%. 实验表明, 提出的跟踪算法可以有效地提高跟踪的整体性能, 实现对目标的持续跟踪.
Abstract:Given that the existing multi-target tracking algorithm cannot track accurately after occlusion, a multi-target tracking algorithm using the improved attention mechanism and Kalman filter is proposed. The structure of joint detection and embedding is used to extract features and accomplish object detection and identification simultaneously. A parallel-structured attention mechanism is proposed, containing both spatial and channel parts. Each part is designed into parallel branches for pooling and convolution. During tracking, the proposed velocity-prediction Kalman filter is adopted for the more accurate prediction of pedestrian movements. The CUHK-SYSU dataset is used for training, and the algorithm is verified and tested on the MOT16 dataset. The proposed algorithm can achieve 65.1% MOTA, 78.8% MOTP, and 62.3% IDF1. The experimental results show that the proposed tracking algorithm can improve the overall tracking performance and achieve continuous tracking.
文章编号:     中图分类号:    文献标志码:
基金项目:国家自然科学基金(61733015); 高铁联合基金(U1934204); 四川省重点研发计划(2020YFQ0057); 四川省自然资源科研项目(KYL202106-0099)
引用文本:
秦泽宇,黄进,杨旭,郑思宇,付国栋.基于注意力机制和卡尔曼滤波的多目标跟踪.计算机系统应用,2021,30(12):128-138
QIN Ze-Yu,HUANG Jin,YANG Xu,ZHENG Si-Yu,FU Guo-Dong.Multi-Target Tracking Using Attention Mechanism and Kalman Filter.COMPUTER SYSTEMS APPLICATIONS,2021,30(12):128-138