计算机科学2021,Vol.48Issue(4) :123-129.DOI:10.11896/jsjkx.200800164

基于时空注意力机制的目标跟踪算法

Object Tracking Algorithm Based on Temporal-Spatial Attention Mechanism

程旭 崔一平 宋晨 陈北京 郑钰辉 史金钢
计算机科学2021,Vol.48Issue(4) :123-129.DOI:10.11896/jsjkx.200800164

基于时空注意力机制的目标跟踪算法

Object Tracking Algorithm Based on Temporal-Spatial Attention Mechanism

程旭 1崔一平 1宋晨 1陈北京 1郑钰辉 1史金钢2
扫码查看

作者信息

  • 1. 南京信息工程大学计算机与软件学院 南京 210044;数字取证教育部工程研究中心南京信息工程大学 南京 210044
  • 2. 西安交通大学软件学院 西安 710049
  • 折叠

摘要

目标跟踪技术在智能监控、人机交互、无人驾驶等诸多领域得到了广泛的应用.近年来,学者们提出了许多高效的算法.然而,随着跟踪环境越来越复杂,目标跟踪算法在遮挡、光照变化、背景干扰等复杂环境下仍然面临着巨大的挑战,从而导致目标跟踪失败.针对上述问题,提出了一种基于时空注意力机制的目标跟踪算法.首先,采用孪生网络架构来提高对特征的判别能力;然后,引入改进的通道注意力机制和空间注意力机制,对不同通道和空间位置的特征施加不同的权重,并着重关注空间位置和通道位置上对目标跟踪有利的特征.此外,还提出了一种高效的目标模板在线更新机制,将第一帧图像特征与后续跟踪图像帧中置信度较高的图像特征进行融合,以降低发生目标漂移的风险.最后,在OTB2013和OTB2015数据集上对所提跟踪算法进行了测试.实验结果表明,所提算法的性能相比当前主流的跟踪算法提高了6.3%.

关键词

深度学习/目标跟踪/孪生网络/注意力机制/模板更新

引用本文复制引用

基金项目

出版年

2021
计算机科学
重庆西南信息有限公司(原科技部西南信息中心)

计算机科学

CSTPCDCSCD北大核心
影响因子:0.944
ISSN:1002-137X
被引量5
参考文献量36
段落导航相关论文