计算机应用 ›› 2020, Vol. 40 ›› Issue (6): 1774-1782.DOI: 10.11772/j.issn.1001-9081.2019112001

• 虚拟现实与多媒体计算 • 上一篇    下一篇

基于相关滤波与颜色概率模型的目标跟踪算法

张杰, 常天庆, 戴文君, 郭理彬, 张雷   

  1. 陆军装甲兵学院 兵器与控制系,北京 100072
  • 收稿日期:2019-11-26 修回日期:2020-01-14 出版日期:2020-06-10 发布日期:2020-06-18
  • 通讯作者: 张杰(1994—)
  • 作者简介:张杰(1994—),男,重庆人,硕士研究生,主要研究方向:目标跟踪.常天庆(1963—),男,河南郑州人,教授,博士,主要研究方向:战车火控系统智能化、模式识别、智能控制.戴文君(1993—),男,湖南衡阳人,博士研究生,主要研究方向:图像处理、模式识别.郭理彬(1981—),男,江西南康人,讲师,硕士,主要研究方向:导航制导与控制.张雷(1994—),男,吉林长春人,副教授,博士,主要研究方向:武器系统、运用工程.

Object tracking algorithm based on correlation filtering and color probability model

ZHANG Jie, CHANG Tianqing, DAI Wenjun, GUO Libin, ZHANG Lei   

  1. Department of Weapon and Control, Army Academy of Armored Forces, Beijing 100072, China
  • Received:2019-11-26 Revised:2020-01-14 Online:2020-06-10 Published:2020-06-18
  • Contact: ZAHNG Jie, born in 1994, M. S. candidate. His research interests include object tracking.
  • About author:ZAHNG Jie, born in 1994, M. S. candidate. His research interests include object tracking.CAHNG Tianqing, born in 1963, Ph. D., professor. His research interests include intelligent technology of combat vehicle fire control system, pattern recognition, intelligent control.DAI Wenjun, born in 1993, Ph. D. candidate. His research interests include image processing, pattern recognition.GUO Libin, born in 1981, M. S., lecturer. His research interests include navigation, guidance and control.ZAHNG Lei, born in 1974, Ph. D., associate professor. His research interests include weapon systems, operation engineering.

摘要: 针对地面战场环境下相似背景对目标跟踪器产生的干扰,提出了一种基于相关滤波与改进颜色概率模型的目标跟踪算法。首先,在传统颜色概率模型的基础上,利用前景目标直方图与背景直方图的差异性提出了一种突出前景的颜色概率模型;然后,根据相关滤波器响应置信度和最大响应位置生成空间惩罚矩阵,用该矩阵惩罚相关滤波器判定的背景像素的似然概率,利用积分图的方法得到颜色概率模型响应图;最后,将相关滤波器和颜色概率模型得到的响应图进行融合,融合响应图的最大响应位置即为目标的中心位置。与核循环结构滤波器(CSK)、核相关滤波器(KCF)、判别式尺度空间跟踪(DSST)、SAMF、Staple等5种算法在跟踪性能上进行比较,在OTB-100标准数据集上的结果表明,所提算法的整体精度提高了3.06%至55.98%,成功率提高了2.24%至54.97%;在相似背景干扰下,其精度提高了10.28%至43.9%,成功率提高了8.3%至48.29%。在36段战场视频序列上的结果表明,所提算法的整体精度提高了2.2%至45.98%,成功率提高了3.01%至58.27%。该算法能够更好地应对地面战场环境下相似背景的干扰,为武器平台提供更精确的位置信息。

关键词: 目标跟踪, 相关滤波, 颜色概率模型, 似然概率, 空间惩罚

Abstract: In order to solve the interference of similar background to object tracker in ground battlefield environment, an object tracking algorithm combining correlation filtering and improved color probability model was proposed. Firstly, based on the traditional color probability model, a color probability model emphasizing foreground was proposed by using the difference between foreground object histogram and background histogram. Then, a spatial penalty matrix was generated according to the correlation filter response confidence and maximum response position. This matrix was used to punish the likelihood probability of background pixel determined by the correlation filter, and the response map of the color probability model was obtained by using the method of integral image. Finally, the response maps obtained by the correlation filter and the color probability model were fused, and the maximum response position of the fusion response map was the central position of the object. The proposed algorithm was compared with 5 state-of-the-art algorithms such as Circulant Structure of tracking-by-detection filters with Kernels (CSK), Kernelized Correlation Filters (KCF), Discriminative Scale Space Tracking (DSST), Scale Adaptive Multiple Feature (SAMF) and Staple in tracking performance. The experimental results on OTB-100 standard dataset show that, the proposed algorithm has the overall accuracy improved by 3.06% to 55.98%, and the success rate improved by 2.24% to 54.97%; and under similar background interference, the proposed algorithm has the accuracy improved by 10.28% to 43.9%, and the success rate improved by 8.3% to 48.29%. The experimental results on 36 battlefield video sequences show that, the proposed algorithm has the overall accuracy improved by 2.2% to 45.98%, and the success rate improved by 3.01% to 58.27%. It can be seen that the proposed algorithm can better deal with the interference of similar background in the ground battlefield environment, and provide more accurate position information for the weapon platform.

Key words: object tracking, correlation filtering, color probability model, likelihood probability, spatial penalty

中图分类号: