计算机工程与应用 ›› 2020, Vol. 56 ›› Issue (19): 132-138.DOI: 10.3778/j.issn.1002-8331.1907-0266

• 模式识别与人工智能 • 上一篇    下一篇

基于多尺度建议框的目标跟踪误差修正方法

张宏伟,李晓霞,朱斌,马旗   

  1. 国防科技大学 电子对抗学院,合肥 230037
  • 出版日期:2020-10-01 发布日期:2020-09-29

Error Refinement Method of Target Tracking Based on Multi-scale Region Proposal

ZHANG Hongwei, LI Xiaoxiang, ZHU Bin, MA Qi   

  1. School of Electronic Countermeasure, National University of Defense Technology, Hefei 230037, China
  • Online:2020-10-01 Published:2020-09-29

摘要:

随着深度学习技术引入视觉目标跟踪领域,目标跟踪算法的精度和鲁棒性有了很大的提高。但在低空无人机跟踪目标的实际场景中,情况比较复杂,如相机的抖动、大量的遮挡、视角和焦距的改变等,使得跟踪算法的准确性受到极大挑战。目前的算法大多建立在目标外观变化缓慢的前提假设下,在跟踪的过程中不具备检测和修复漂移(跟踪误差)的能力。针对该问题,提出了一种基于多尺度建议框的目标跟踪误差修正方法。离线阶段,利用大量的已标注的目标样本训练基于多尺度建议框的目标跟踪修正模型,获取不同类别目标的先验知识。在线阶段在核相关滤波跟踪的基础上,依据相关响应置信度自适应评价的结果,通过目标跟踪修正模型不定期重新初始化目标的位置,避免了因为误差累积而导致跟踪失败。算法在无人机航拍数据集上进行了测试,结果表明,该跟踪算法在目标发生较大形变的情况下能较好的修正跟踪漂移问题。相比于其他几种算法,目标跟踪的成功率和精度分别提高了14.3%和3.1%。

关键词: 视觉跟踪, 卷积神经网络, 核相关滤波, 跟踪漂移, 误差累积

Abstract:

With the introduction of deep learning technology into the field of visual target tracking, the accuracy and robustness of the target tracking have greatly improved. However, in the actual scene of target tracking with low altitude UAV, the complex situations, such as camera jitter, a large number of occlusions, changes of shooting perspective and focal length, etc. , make the tracking algorithm face great challenges. Most of recent algorithms are based on the assumption that the appearance of the target changes slowly; they do not have the ability to detect and repair the drift(tracking error)in the tracking process. In order to solve this problem, an error refinement method of target tracking based on multi-scale region proposal is proposed. In the offline stage, a large number of labeled target samples are used to train the tracking refinement model based on multi-scale region proposal method, which can obtain prior knowledge of different kinds of targets. In the online stage, kernel correlation filter-based tracking algorithm is used to track the target in real time. According to the self-adaptive evaluation of the correlation response confidence, the tracking refinement model is used to initialize the position of target irregularly, so as to avoid the loss of target due to the error accumulation. Experimental results on UAV aerial datasets show that the proposed algorithm can correct the tracking drift problem accurately when targets undergo heavy deformation. Compared with the state-of-art algorithms, the success rate and the precision of target tracking are improved by 14.3% and 3.1%, respectively.

Key words: visual tracking, convolutional neural network, kernel correlation filter, tracking drift, cumulative error