留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

融合通道互联空间注意力的Siamese网络跟踪算法

崔洲涓 安军社 崔天舒

崔洲涓, 安军社, 崔天舒. 融合通道互联空间注意力的Siamese网络跟踪算法[J]. 红外与激光工程, 2021, 50(3): 20200148. doi: 10.3788/IRLA20200148
引用本文: 崔洲涓, 安军社, 崔天舒. 融合通道互联空间注意力的Siamese网络跟踪算法[J]. 红外与激光工程, 2021, 50(3): 20200148. doi: 10.3788/IRLA20200148
Cui Zhoujuan, An Junshe, Cui Tianshu. Siamese networks tracking algorithm integrating channel-interconnection-spatial attention[J]. Infrared and Laser Engineering, 2021, 50(3): 20200148. doi: 10.3788/IRLA20200148
Citation: Cui Zhoujuan, An Junshe, Cui Tianshu. Siamese networks tracking algorithm integrating channel-interconnection-spatial attention[J]. Infrared and Laser Engineering, 2021, 50(3): 20200148. doi: 10.3788/IRLA20200148

融合通道互联空间注意力的Siamese网络跟踪算法

doi: 10.3788/IRLA20200148
基金项目: 中国科学院复杂航天系统电子信息技术重点实验室自主部署基金(Y42613A32S)
详细信息
    作者简介:

    崔洲涓(1986−),女,计算机应用技术专业博士生,主要从事计算机视觉、深度学习、目标跟踪等方面的研究

    安军社(1969−),男,研究员、博士生导师,计算机软件与理论专业博士。以第一作者在核心刊物上发表论文8篇,其他刊物和会议论文6篇。申请专利16项,其中以第一发明人获得发明专利4项,已获专利4项,作为合作者获得专利超过30项。参加载人航天、探月工程、中俄联合火星探测、北斗导航和自主火星探测等多项国家重点项目。主要研究领域为空间飞行器综合电子系统,主要研究方向为星载计算平台、星载数据网络技术等。获部委奖励3次。曾获科学技术进步奖三等奖,获部委科技进步一等奖;获科工委颁发的"绕月探测工程初样研制建设先进个人"称号,中国科学院颁发的"参加载人航天工程优秀工作者"荣誉称号,中国科学院空间科学与应用总体部颁发的"神舟2号飞船应用系统任务先进个人";1999年11月,载人航天第一次飞行试验突出贡献奖成员。主要从事航天计算器等方面的研究

  • 中图分类号: TP391

Siamese networks tracking algorithm integrating channel-interconnection-spatial attention

  • 摘要: 基于Siamese网络的跟踪算法在跟踪精度和速度方面展现出巨大的潜力,然而要使离线训练的模型适应在线跟踪仍然面临着挑战。为了提升复杂场景下算法的特征提取以及判别能力,提出了一种融合通道-互联-空间注意力的Siamese网络实时跟踪算法。首先构建以深度卷积网络VGG-Net-16作为主干网络的Siamese跟踪框架,增加特征提取能力;接着设计通道-互联-空间注意力模块,增强模型的适应能力与判别能力;然后加权融合多层响应图,获取更精准的跟踪结果;最后使用大规模数据集对网络进行端到端的训练,在通用数据集OTB-2015上进行跟踪测试。实验结果表明:与当前主流算法相比,所提算法具有较强的稳健性,能更好地适应目标外观变化、相似物干扰、目标遮挡等复杂场景,在NVIDIA RTX 2060 GPU上,跟踪速度平均达到37FPS,满足实时性要求。
  • 图  1  Siamese网络结构图

    Figure  1.  Framework of Siamese network

    图  2  算法框架图

    Figure  2.  Framework of algorithm

    图  3  VGG-Net-16提取的部分特征图

    Figure  3.  A set of feature maps extracted by VGG-Net-16

    图  4  CISAM结构图

    Figure  4.  Structure of CISAM

    图  5  Grad-CAM网络可视化结果

    Figure  5.  Grad-CAM network visualization results

    图  6  六种跟踪算法在部分视频序列上的定性结果显示

    Figure  6.  Qualitative results display of the 6 tracking algorithms over partial video sequences

    图  7  基于OTB-2015 OPE的跟踪精度曲线图

    Figure  7.  Tracking precision plots of OPE on OTB-2015 dataset

    图  8  基于OTB-2015 OPE的成功率曲线图

    Figure  8.  Success plots of OPE on OTB-2015 dataset

    图  9  OTB-2015 11种不同属性视频序列跟踪精度曲线

    Figure  9.  Tracking precision plots of 11 different attributes video sequences on OTB-2015 dataset

    图  10  OTB-2015 11种不同属性视频序列成功率曲线

    Figure  10.  Success plots of 11 different attributes video sequences on OTB-2015 dataset

    表  1  Siamese网络结构

    Table  1.   Architecture of Siamese network

    Layer nameKernel sizeChan. mapTemplate sizeSearch sizeChannel outputStrideCISAM
    Input127×127255×2553-No
    Conv1_13×364×3125×125253×253641No
    Conv1_23×364×64123×123251×251641No
    Pool12×261×61125×125642No
    Conv2_13×3128×6459×59123×1231281No
    Conv2_23×3128×12857×57121×1211281Yes
    Pool22×228×2860×601282No
    Conv3_13×3256×12826×2658×582561No
    Conv3_23×3256×25624×2456×562561No
    Conv3_33×3256×25622×2254×542561Yes
    Pool32×211×1127×272562No
    Conv4_13×3512×2569×925×255121No
    Conv4_23×3512×5127×723×235121No
    Conv4_33×3512×5125×521×215121Yes
    下载: 导出CSV

    表  2  实验平台参数

    Table  2.   Experimental platform parameters

    DeviceProduct modelMemory
    CPUIntel(R)Core(TM)i7-970016G
    Basic frequency 3.0 GHz
    GPUNVIDIA GeForce RTX-20606G
    下载: 导出CSV
  • [1] Ma C, Huang J B, Yang X, et al. Hierarchical convolutional features for visual tracking[C]//Proceedings of the IEEE International Conference on Computer Vision, 2015: 3074−3082.
    [2] Danelljan M, Robinson A, Khan F S, et al. Beyond correlation filters: Learning continuous convolution operators for visual tracking[C]//ECCV, 2016.
    [3] Danelljan M, Bhat G, Khan F S, et al. Eco: Efficient convolution operators for tracking[C]//CVPR, 2017.
    [4] Tao Ran, Gavves E, Smeulders A W M. Siamese instance search for tracking[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition, 2016: 1420−1429.
    [5] Bertinetto L, Valmadre J, Henriques J F, et al. Fully-Convolutional Siamese Networks for Object Tracking[M]//Hua G, Jegou H. Computer Version ECCV 2016 Workshops. Cham: Springer, 2016, 9914: 850−865.
    [6] Valmadre J, Bertinetto L, Henriques J F, et al. End-to-end representation learning for correlation filter based tracking[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition, 2017: 5000−5008.
    [7] Li Bo, Yan Junjie, Wu Wei, et al. High performance visual tracking with Siamese region proposal network[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2018: 8971−8980.
    [8] Zhu Zheng, Wang Qiang, Li Bo, et al. Distractor-aware Siamese networks for visual object tracking[C]//The 15th European Conference on Computer Vision, 2018: 103−119.
    [9] Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition[EB/OL]. (2015-04-10)[2018-12-15]. https://arxiv.org/abs/1409.1556.
    [10] Bromley J, Guyon I, LeCun Y, et al. Signature verification using a “Siamese” time delay neural network[C]//Advances in Neural Information Processing Systems, 1994: 737−744.
    [11] Zagoruyko S, Komodakis N. Learning to compare image patches via convolutional neural networks[C]//CVPR, 2015.
    [12] Wang N, Shi J, Yeung D, et al. Understanding and Diagnosing Visual Tracking Systems[C]//2015 IEEE International Conference on Computer Vision (ICCV), 2015: 3101−3109.
    [13] 黎万义, 王鹏, 乔红. 引入视觉注意机制的目标跟踪方法综述[J]. 自动化学报, 2014, 40(4): 561-576.

    Li Wanyi, Wang Peng, Qiao Hong. A survey of visual attention based methods for object tracking [J]. Acta automatica sinica, 2014, 40(4): 561-576. (in Chinese)
    [14] Woo S, Park J, Lee J Y, etal. CBAM: Convolutional Block Attention Module[C]// European Conference on Computer Vision, 2018: 3−19.
    [15] Russakovsky O, Deng J, Su H, et al. Image net large scale visual recognition challenges[C]//IJCV, 2015.
    [16] Lin T-Y, Maire M, Belongie S, et al. Microsoft coco: Common objects in context[C]//ECCV, 2014: 740−755.
    [17] Real E, Shlens J, Mazzocchi S, et al. Youtube boundingboxes: A large high-precision human-annotated data set for object detection in video[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), IEEE, 2017: 7464−7473.
    [18] Selvaraju R R, Cogswell M, Das A, et al. Grad-cam: Visual explanations from deep networks via gradient-based localization[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2017: 618−626.
    [19] Wu Y, Lim J, Yang M H. Object tracking benchmark [J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9): 1834-1848. doi:  10.1109/TPAMI.2014.2388226
  • [1] 夏文新, 杨小冈, 席建祥, 卢瑞涛, 谢学立.  基于红外探测的无人机群结构特性感知方法 . 红外与激光工程, 2024, 53(1): 20230429-1-20230429-12. doi: 10.3788/IRLA20230429
    [2] 李鹏越, 续欣莹, 唐延东, 张朝霞, 韩晓霞, 岳海峰.  基于并行多轴自注意力的图像去高光算法 . 红外与激光工程, 2024, 53(3): 20230538-1-20230538-11. doi: 10.3788/IRLA20230538
    [3] 王向军, 朱慧.  利用国产FPGA的高帧率目标跟踪方法 . 红外与激光工程, 2023, 52(9): 20220905-1-20220905-12. doi: 10.3788/IRLA20220905
    [4] 张景程, 乔新博, 赵永强.  红外偏振摄像机动目标检测跟踪系统(特邀) . 红外与激光工程, 2022, 51(4): 20220233-1-20220233-10. doi: 10.3788/IRLA20220233
    [5] 许聪, 孙大鹰, 曹子奇, 李春琦, 顾文华.  采用自适应模糊优化的目标跟踪加速方案 . 红外与激光工程, 2022, 51(2): 20210864-1-20210864-8. doi: 10.3788/IRLA20210864
    [6] 薛珊, 陈宇超, 吕琼莹, 曹国华.  基于坐标注意力机制融合的反无人机系统图像识别方法 . 红外与激光工程, 2022, 51(9): 20211101-1-20211101-11. doi: 10.3788/IRLA20211101
    [7] 赵晓枫, 徐叶斌, 吴飞, 牛家辉, 蔡伟, 张志利.  基于并行注意力机制的地面红外目标检测方法(特邀) . 红外与激光工程, 2022, 51(4): 20210290-1-20210290-8. doi: 10.3788/IRLA20210290
    [8] 张津浦, 王岳环.  融合检测技术的孪生网络跟踪算法综述 . 红外与激光工程, 2022, 51(10): 20220042-1-20220042-14. doi: 10.3788/IRLA20220042
    [9] 张骏, 朱标, 沈玉真, 张鹏.  基于引导滤波的多分支注意力残差红外图像去噪网络 . 红外与激光工程, 2022, 51(11): 20220060-1-20220060-11. doi: 10.3788/IRLA20220060
    [10] 王向军, 欧阳文森.  多尺度循环注意力网络运动模糊图像复原方法 . 红外与激光工程, 2022, 51(6): 20210605-1-20210605-9. doi: 10.3788/IRLA20210605
    [11] 张宏伟, 李晓霞, 朱斌, 张杨.  基于孪生神经网络的两阶段目标跟踪方法 . 红外与激光工程, 2021, 50(9): 20200491-1-20200491-12. doi: 10.3788/IRLA20200491
    [12] 葛宝义, 左宪章, 胡永江, 张岩.  基于双步相关滤波的目标跟踪算法 . 红外与激光工程, 2018, 47(12): 1226004-1226004(10). doi: 10.3788/IRLA201847.1226004
    [13] 马俊凯, 罗海波, 常铮, 惠斌, 周晓丹, 侯德飞.  基于可变形模型的目标跟踪算法 . 红外与激光工程, 2017, 46(9): 928001-0928001(9). doi: 10.3788/IRLA201746.0928001
    [14] 罗海波, 许凌云, 惠斌, 常铮.  基于深度学习的目标跟踪方法研究现状与展望 . 红外与激光工程, 2017, 46(5): 502002-0502002(7). doi: 10.3788/IRLA201746.0502002
    [15] 邢运龙, 李艾华, 崔智高, 方浩.  改进核相关滤波的运动目标跟踪算法 . 红外与激光工程, 2016, 45(S1): 214-221. doi: 10.3788/IRLA201645.S126004
    [16] 邵春艳, 丁庆海, 罗海波, 李玉莲.  采用高维数据聚类的目标跟踪 . 红外与激光工程, 2016, 45(4): 428002-0428002(10). doi: 10.3788/IRLA201645.0428002
    [17] 崔雄文, 吴钦章, 蒋平, 周进.  子空间模型下的仿射不变目标跟踪 . 红外与激光工程, 2015, 44(2): 769-774.
    [18] 肖松, 谭贤四, 王红, 李志淮.  变结构多模型临近空间高超声速飞行器跟踪算法 . 红外与激光工程, 2014, 43(7): 2362-2370.
    [19] 黎志华, 李新国.  基于OpenCV的红外弱小运动目标检测与跟踪 . 红外与激光工程, 2013, 42(9): 2561-2565.
    [20] 彭晨, 陈钱, 钱惟贤, 徐富元.  复杂地面场景下的红外运动目标跟踪 . 红外与激光工程, 2013, 42(6): 1410-1414.
  • 加载中
图(10) / 表(2)
计量
  • 文章访问数:  466
  • HTML全文浏览量:  168
  • PDF下载量:  44
  • 被引次数: 0
出版历程
  • 收稿日期:  2020-04-26
  • 修回日期:  2020-06-02
  • 网络出版日期:  2021-05-12
  • 刊出日期:  2021-03-15

融合通道互联空间注意力的Siamese网络跟踪算法

doi: 10.3788/IRLA20200148
    作者简介:

    崔洲涓(1986−),女,计算机应用技术专业博士生,主要从事计算机视觉、深度学习、目标跟踪等方面的研究

    安军社(1969−),男,研究员、博士生导师,计算机软件与理论专业博士。以第一作者在核心刊物上发表论文8篇,其他刊物和会议论文6篇。申请专利16项,其中以第一发明人获得发明专利4项,已获专利4项,作为合作者获得专利超过30项。参加载人航天、探月工程、中俄联合火星探测、北斗导航和自主火星探测等多项国家重点项目。主要研究领域为空间飞行器综合电子系统,主要研究方向为星载计算平台、星载数据网络技术等。获部委奖励3次。曾获科学技术进步奖三等奖,获部委科技进步一等奖;获科工委颁发的"绕月探测工程初样研制建设先进个人"称号,中国科学院颁发的"参加载人航天工程优秀工作者"荣誉称号,中国科学院空间科学与应用总体部颁发的"神舟2号飞船应用系统任务先进个人";1999年11月,载人航天第一次飞行试验突出贡献奖成员。主要从事航天计算器等方面的研究

基金项目:  中国科学院复杂航天系统电子信息技术重点实验室自主部署基金(Y42613A32S)
  • 中图分类号: TP391

摘要: 基于Siamese网络的跟踪算法在跟踪精度和速度方面展现出巨大的潜力,然而要使离线训练的模型适应在线跟踪仍然面临着挑战。为了提升复杂场景下算法的特征提取以及判别能力,提出了一种融合通道-互联-空间注意力的Siamese网络实时跟踪算法。首先构建以深度卷积网络VGG-Net-16作为主干网络的Siamese跟踪框架,增加特征提取能力;接着设计通道-互联-空间注意力模块,增强模型的适应能力与判别能力;然后加权融合多层响应图,获取更精准的跟踪结果;最后使用大规模数据集对网络进行端到端的训练,在通用数据集OTB-2015上进行跟踪测试。实验结果表明:与当前主流算法相比,所提算法具有较强的稳健性,能更好地适应目标外观变化、相似物干扰、目标遮挡等复杂场景,在NVIDIA RTX 2060 GPU上,跟踪速度平均达到37FPS,满足实时性要求。

English Abstract

    • 作为计算机视觉的研究方向之一,目标跟踪在视频监控、智能交通、军事制导、航天航空等领域都有广泛的应用。然而由于目标自身形变、旋转、运动模糊,以及外部应用场景光照变化、背景干扰、遮挡等因素的影响,建立一个高效稳健的目标跟踪算法依然面临着巨大的挑战。

      随着深度学习方法在图像分类、目标检测等领域的应用取得了突破性的进展,研究者也逐步将其引入目标跟踪领域,利用深度卷积网络出色的特征表达能力提升算法的精度。HCF[1]利用预训练神经网络提取深层和浅层特征,并根据不同层的特点,分别训练相关滤波器,得到响应图加权融合,提升了跟踪精度。C-COT[2]通过连续空间域插值转换,将预训练神经网络提取的不同分辨率的特征图插值到连续空域,结合多分辨率的连续卷积滤波器进行训练和检测,取得了更加精确的定位。ECO[3]从滤波器系数、样本划分和模板更新策略三个方面进行优化,在保持跟踪精度的同时,大幅度提高了跟踪速度。相比传统手工特征,这类算法利用预训练深度特征结合相关滤波框架,可以提升算法的跟踪精度。但预训练深度模型大多是针对图像分类任务而设计,并未能较好地适应目标跟踪。同时,越来越高的特征维度,给在线学习、更新过程带来了巨大的计算开销,直接限制了算法的跟踪速度。

      深度学习方法在目标跟踪领域的应用不仅局限于预训练深度特征,为了利用端到端的优势,研究者们开始引入Siamese框架,训练专门的端到端跟踪网络。SINT[4]开创性地引入Siamese网络,将目标跟踪任务转化为相似性学习问题,学习一个匹配函数。在跟踪过程中将后续帧的候选框与第一帧目标框进行匹配度计算,得分最高的即为目标。通过使用涵盖了各种目标变化可能情况的大规模训练数据集离线训练模板匹配网络,在不应用任何模型更新,没有遮挡检测,没有跟踪器的组合,没有几何匹配的情况下实现目标跟踪。SiamFC[5]针对跟踪任务使用大规模视觉图像识别比赛中的视频离线训练一个深度网络,跟踪过程对候选图像进行全卷积,通过计算其两个输入互相关的双线性层完成滑窗在线评估,在保证高准确率的同时,速度也远超实时性。CFNet[6]将相关滤波转换为可微分的神经网络层,联合特征提取网络实现端到端的优化,训练与相关滤波器匹配的卷积特征,学习变换背景矩阵与目标外观矩阵,达到适应目标变化以及抑制背景变化的目的。SiamRPN[7]引入区域候选网络通过边界框回归代替多尺度检测,网络结构包括特征提取的 Siamese子网络和产生候选目标区域的RPN子网络。通过将Siamese子网络模板分支、检测分支的特征同时输入到RPN子网络的分类分支与回归分支,将跟踪问题转换成为单次学习匹配问题。DasiamRPN[8]从训练数据不平衡、自适应的模型增量学习及长时跟踪等方面对算法进行了优化,在目标遮挡和长时跟踪等情况下表现突出。

      尽管基于Siamese网络的端到端跟踪算法在通用数据集中取得了卓越的跟踪效果,但仍存在一些问题。离散训练网络只学习到了目标的通用特征,若在线跟踪过程中目标周围出现相似干扰物,则无法进行判断。在线跟踪的目标很可能并未包含在离线训练数据集中,那么相似性度量的结果未必可靠。由于特征提取与判别功能耦合在同一个网络里,缺少了分类器的在线训练环节,导致网络的适应能力、判别能力较弱。文中提出一种端到端的目标跟踪算法,承继Siamese网络架构的特点,选取更深的卷积网络模型VGG-Net-16[9]作为主干网络,使模型具有更强的特征表达能力;探索多种注意力机制,包括通道注意力、互联注意力以及空间注意力,使离线训练的模型在线跟踪时具有更优越的适应能力以及判别能力;通过加权融合响应得分图,以获取更精准的定位;在大规模数据集上离线训练端到端网络。在目标跟踪通用数据集中,进行算法性能的验证,与当前主流算法相比,文中算法在保证实时性的前提下,能够取得良好的跟踪精度。

    • Siamese网络结构[10]最早提出用于验证支票上的签名是否与银行预留的签名一致,后来Sergey Zagoruyko[11]等将其引入进行图像相似度的计算。

      图1所示,Siamese网络结构通常由结构相似、权值共享的两个分支构成。将两个图片${{{X}}_1}$${{{X}}_2}$分别输入两个分支,通过构建一种函数${{{G}}_{\rm{w}}}\left( {{X}} \right)$将其映射到特征空间,得到两个特征向量${{{G}}_{\rm{w}}}\left( {{{{X}}_1}} \right)$${{{G}}_{\rm{w}}}\left( {{{{X}}_2}} \right)$,使用一种距离度量${{{E}}_{\rm{w}}}\left( {{{{X}}_1},{{{X}}_2}} \right)$,作为两张图片的相似度计算函数,衡量二者之间的差异。

      图  1  Siamese网络结构图

      Figure 1.  Framework of Siamese network

    • 近年来,在深度学习的发展推动下,Siamese网络被应用于目标跟踪领域,并取得了令人瞩目的成绩。基于Siamese网络的代表性跟踪算法SiamFC[5]提出,将目标跟踪任务转化为相似性度量问题,利用大量样本数据离线训练一个Siamese网络(由卷积层、非线性激活层和池化层构成),将视频序列第一帧作为模板分支的输入图像${{\textit{z}}}$,后续帧中的候选区域作为检测分支的输入图像${{x}}$,分别通过权值共享的特征提取网络$\varphi \left( \cdot \right)$,将原始图像映射到特定的特征空间,学习一个度量函数$f(z,x)$,如公式(1)所示,用于比较模板图像和候选区域搜索图像之间的相似度,返回响应图,分数越高,二者相似度越高。其中$ * $代表互相关运算,$b \cdot {{I}}$表示在响应图中每个位置的取值。

      $$ f({{\textit{z}}},{{x}}) = \varphi \left( {{\textit{z}}} \right) * \varphi \left( {{x}} \right) + b \cdot {{I}} $$ (1)
    • 算法框架如图2所示。

      图  2  算法框架图

      Figure 2.  Framework of algorithm

      主要包括Siamese深度网络、通道-互联-空间注意力模块(Channel-Interconnection-Spatial Attention Module,CISAM)、互相关运算以及响应得分图融合四部分组成。其中Siamese深度网络基于VGG-Net-16进行构建,再融入CISAM。图像序列经过改进后的Siamese网络,将原始图像映射到特定的特征空间,得到多层特征图,分别进行互相关运算,输出响应得分图(Response Score Map)进行融合,得分越高,则相应位置是目标的概率越大。

    • 特征提取是决定跟踪算法性能的最关键因素[12]。为了提升网络的特征提取能力,而又不引入填充破坏网络的平移不变性,文中网络基于适应能力更强的VGG-Net-16进行构建。

      VGG-Net-16是由3×3的小型卷积核以及2×2的最大池化层反复堆叠构筑而成的卷积神经网络,通过小型卷积核的深度复用模拟较大卷积核完成对图像的局部感知,提升网络的性能。模型主要由5段卷积、2个全连接特征层以及1个全连接分类层组成。VGG-Net-16能够提取到非常丰富的特征,将其提取到的不同层不同通道的部分特征图可视化,如图3所示。

      图  3  VGG-Net-16提取的部分特征图

      Figure 3.  A set of feature maps extracted by VGG-Net-16

      为了将VGG-Net-16更好地运用到文中所提算法中,根据Siamese网络的特点,同时考虑到后续互相关以及响应图的融合等操作,对VGG-Net-16进行修改,具体网络结构如表1所示。从表中可以看出,修改主要包括以下四个方面。

      (1)删减:考虑到目标跟踪任务对精细位置的需求,删除第五段卷积,缩小了网络模型的总步长;

      (2)增加:在网络模型内,在卷积层Conv2_2、Conv3_3、Conv4_3后均融入了通道-互联-空间注意力模块CISAM,形成特征图层Conv_CISAM,尺寸维持不变,分别记作Conv2_2_CISAM、Conv3_3_CISAM以及Conv4_3_CISAM;

      (3)提取:为了利用多分支特征协同进行定位,将Conv2_2_CISAM、Conv3_3_CISAM以及Conv4_3_CISAM分别提取;

      (4)调整:为了便于后续响应得分图计算时的融合,在进行互相关运算前,将多分支特征Conv2_2_CISAM、Conv3_3_CISAM以及Conv4_3_CISAM分别输入调整层Conv_adjust进行调节。

      表 1  Siamese网络结构

      Table 1.  Architecture of Siamese network

      Layer nameKernel sizeChan. mapTemplate sizeSearch sizeChannel outputStrideCISAM
      Input127×127255×2553-No
      Conv1_13×364×3125×125253×253641No
      Conv1_23×364×64123×123251×251641No
      Pool12×261×61125×125642No
      Conv2_13×3128×6459×59123×1231281No
      Conv2_23×3128×12857×57121×1211281Yes
      Pool22×228×2860×601282No
      Conv3_13×3256×12826×2658×582561No
      Conv3_23×3256×25624×2456×562561No
      Conv3_33×3256×25622×2254×542561Yes
      Pool32×211×1127×272562No
      Conv4_13×3512×2569×925×255121No
      Conv4_23×3512×5127×723×235121No
      Conv4_33×3512×5125×521×215121Yes
    • 仅仅增加网络的深度只能相对提升特征表达能力,并不能从根本上解决Siamese框架存在的问题。因为相对于相关滤波类的跟踪算法,基于Siamese网络的跟踪算法缺少了分类器的在线训练环节,在同一个网络中耦合了特征提取与判别功能。这需要网络同时具备两种特质:一是能够稳定地适应目标自身在各种场景中的变化,抽象出目标代表性、本质性的特征;二是能够敏感地区分目标及相似物,显著地提炼出二者之间的差异。

      设经过Siamese主干网络提取到的特征图分别为$\varphi \left( {{\textit{z}}} \right) \in {\mathbb{R}^{C \times {H_{\rm{T}} } \times {W_{\rm{T}} }}}$$\varphi \left( {{x}} \right) \in {\mathbb{R}^{C \times {H_{\rm{D}} } \times {W_{\rm{D}} }}}$,其中,$C$为通道数,${H_{\rm{T}}}$${H_{\rm{D}}}$为特征图在垂直方向的尺寸,${W_{\rm{T}}}$${W_{\rm{D}}}$为特征图在水平方向的尺寸,${H_{\rm{T}} } \leqslant {H_{\rm{D}} }$${W_{\rm{T}} } \leqslant {W_{\rm{D}} }$,二者进行互相关运算得到的$f({{\textit{z}}},{{x}}) \in {\mathbb{R}^{C \times {H_f} \times {W_f}}}$,其中响应图垂直方向、水平方向的尺寸分别为${H_f} = {H_{\rm{D}} } - {H_{\rm{T}} } + 1$${W_f} = $$ {W_{\rm{D}} } - {W_{\rm{T}} } + 1$$b$为响应图中每个位置的权值。将Siamese网络的匹配函数公式(1)展开如公式(2)所示:

      $$f({{\textit{z}}},{{x}}) = \sum\limits_{c = 0}^{C - 1} {\sum\limits_{h = 0}^{{H_{{\rm{T}}} } - 1} {\sum\limits_{w = 0}^{{W_{{\rm{T}}} } - 1} {{\varphi _{c,h,w}}\left( {{\textit{z}}} \right){\varphi _{c,{H_f} + h,{W_f} + w}}\left( {{x}} \right)} } } + b $$ (2)

      可以看到,互相关的计算过程对特征图的通道层面、空间层面的关注度非常平均,权重系数均为1。事实上在跟踪任务中,不同通道、不同位置的特征重要性是不同的。研究表明[13],注意力机制在人类视觉过程中起到重要作用,图像中的高层语义特征能够吸引人类的视觉注意力。因此,可以借鉴性地引入注意力机制弱化贡献度小的特征图,强化贡献度大的特征图,关注目标前景和语义背景的差异特征,实现对目标的增强、对干扰的抑制以及对不同对象间的判别,提高复杂场景下算法的稳健性与实时性。为此,选择在特征提取主干网络的卷积层中融入注意力模块,通过调节权重参数突出或筛选目标的重要信息,抑制无关的细节信息,提升网络的判别能力,获得相当于分类器在线学习的功能。

      CBAM[14]提出将注意力机制同时运用在通道和空间两个层面,可以嵌入目前大部分主流深度网络,在不显著增加计算量和参数量的前提下能提升网络模型的特征提取能力。考虑到Siamese网络的结构特点,文中在此基础上,加入了互联注意力模块,形成通道-互联-空间注意力模块CISAM,具体结构如图4所示。

      图  4  CISAM结构图

      Figure 4.  Structure of CISAM

      模板分支、检测分支卷积层输出的特征图$\varphi \left( {{\textit{z}}} \right)$$\varphi \left( {{x}} \right)$,经过通道注意力模块得到注意图${{{A}}_{\rm C}}\left\{ \cdot \right\} \in {\mathbb{R}^{C \times 1 \times 1}}$,通过公式(3)、(4)分别输出通道特征图${\varphi _{{\rm C}} }\left( {{\textit{z}}} \right)$${\varphi _{{\rm C}} }\left( {{x}} \right)$$ \otimes $表示元素点乘。

      $$ {\varphi _{{\rm C}} }\left( {{\textit{z}}} \right) = {{{A}}_{{\rm C}} }\{ \varphi \left( {{\textit{z}}} \right)\} \otimes \varphi \left( {{\textit{z}}} \right) $$ (3)
      $$ {\varphi _{{\rm C}} }\left( {{x}} \right) = {{{A}}_{{\rm C}} }\{ \varphi \left( {{x}} \right)\} \otimes \varphi \left( {{x}} \right) $$ (4)

      通道注意力模块对输入特征图在空间维度进行压缩,同时使用全局平均池化与全局最大池化,前者对特征图上的每一个像素点都进行反馈,而后者对全局平均池化进行补充。每个通道等价为一个不同类型的特征判别器,从语义层面对特征优化选择,激活与目标相关程度更高的,删除冗余的通道特征。

      同时,模板分支、检测分支卷积层输出的特征图$\varphi \left( {{\textit{z}}} \right)$$\varphi \left( {{x}} \right)$,经过互联注意力模块得到注意图${{{A}}_{\rm I}}\left\{ \cdot \right\} \in {\mathbb{R}^{C \times C}}$,由于两个分支的空间尺寸并不相同,进行互联时需要先调整维度,得到${\varphi _{{\rm{RS}} }}\left( {{\textit{z}}} \right) \in {\mathbb{R}^{C \times {P_{\rm{T}} }}}$${\varphi _{{\rm{RS}} }}\left( {{x}} \right) \in {\mathbb{R}^{C \times {P_{\rm{D}} }}}$,其中${P_{\rm{T}} } = {H_{\rm{T}} } \times {W_{\rm{T}} }$${P_{\rm{D}} } = {H_{\rm{D}} } \times {W_{\rm{D}} }$,通过公式(5)、(6)分别输出互联特征图${\varphi _{{\rm{I}} {\rm{RS}} }}\left( {{\textit{z}}} \right)$${\varphi _{{\rm{I}} {\rm{RS}} }}\left( {{x}} \right)$。再调整恢复维度后输出互联特征图${\varphi _{\rm{I}} }\left( {{\textit{z}}} \right)$${\varphi _{\rm{I}} }\left( {{x}} \right)$

      $$ {\varphi _{{\rm{I}} {\rm{RS}} }}\left( {{\textit{z}}} \right) = {{{A}}_{\rm{I}} }\{ {\varphi _{{\rm{RS}} }}\left( {{x}} \right)\} \otimes {\varphi _{{\rm{RS}} }}\left( {{\textit{z}}} \right) $$ (5)
      $$ {\varphi _{{\rm{I}} {\rm{RS}} }}\left( {{x}} \right) = {{{A}}_{\rm{I}} }\{ {\varphi _{{\rm{RS}} }}\left( {{\textit{z}}} \right)\} \otimes {\varphi _{{\rm{RS}} }}\left( {{x}} \right) $$ (6)

      互联注意力模块将两个分支分别编码至另外一个分支,可以帮助结合目标及其背景之间的相关性。两个分支的通道注意力与互联注意力分别汇合后,再经由空间注意力模块得到注意图${{{A}}_{\rm S}}\left\{ \cdot \right\} \in {\mathbb{R}^{1 \times {H_{\rm{T}} } \times {W_{\rm{T}} }}}$${{{A}}_{\rm S}}\left\{ \cdot \right\} \in {\mathbb{R}^{1 \times {H_{\rm{D}} } \times {W_{\rm{D}} }}}$,通过公式(7)、(8)融合提取出特征图${\varphi _{\rm{S}} }\left( {{\textit{z}}} \right)$${\varphi _{\rm{S}} }\left( {{x}} \right)$

      $${\varphi _{{\rm S}} }\left( {{\textit{z}}} \right) = {{{A}}_{{\rm S}} }\{ \left[ {{\varphi _{{\rm C}} }\left( {{\textit{z}}} \right) + {\varphi _{\rm{I}} }\left( {\bf{\textit{z}}} \right)} \right]\} \otimes \left[ {{\varphi _{{\rm C}} }\left( {{\textit{z}}} \right) + {\varphi _{\rm{I}} }\left( {{\textit{z}}} \right)} \right] $$ (7)
      $${\varphi _{{\rm S}} }\left( {{x}} \right) = {{{A}}_{{\rm S}} }\{ \left[ {{\varphi _{{\rm C}} }\left( {{x}} \right) + {\varphi _{\rm{I}} }\left( {{x}} \right)} \right]\} \otimes \left[ {{\varphi _{{\rm C}} }\left( {{x}} \right) + {\varphi _{\rm{I}} }\left( {{x}} \right)} \right] $$ (8)

      空间注意力模块使用全局平均池化与最大池化对输入特征图在通道层面进行压缩,得到两个二维的特征图,并拼接得到一个通道数为2的特征图,经由隐藏层对其进行卷积操作,保证特征图在空间维度上保持不变。空间注意力更集中于位置的描述,与通道注意力的相互补充,通过构建特征图中不同位置之间的联系,从空间层面学习特征图的哪些部分应该会有更高的响应,针对位置进行加权融合。

    • 充分利用不同层次的特征图,能够提升跟踪性能。浅层特征空间分辨率较高,纹理信息全面,易于定位目标,能够防止表观相似的干扰背景。深层特征语义信息丰富,能够稳定适应目标形变、遮挡等表观变化。为了降低复杂场景中干扰源的影响,获取高质量的响应得分图,采用多层特征融合的方法。首先将CISAM输出的特征图通过调整层将通道数统一调整为256;接着分别计算模板分支与检测分支中各调整层输出的第$q$层特征进行互相关运算;最后将响应得分图以加权的方式进行融合,如公式(9),其中${\alpha _q}$为第$q$层的权重参数。综合分析响应得分图,确定参数的具体值。

      $$ {S_{{\rm{RM}} }} = \sum\limits_{q = 2,3,4} {{\alpha _q}S_{{\rm{RM}} }^{(q)}} $$ (9)
    • (1) 数据集的准备

      训练集中的数据选自ImageNet VID[15]、COCO[16]和Youtube-bb[17],通过特定的比例组合。从相同的视频序列中随机选取两帧,并将它们组合成一对模板图像(127×127)和搜索图像(255×255),作为Siamese网络的输入。

      为保证所有的输入图像均为正方形,按照公式(10)调整,其中$\left( {w,h} \right)$为原始目标框尺寸,$p$为上下文边距,$s$为尺度变换因子,$A = {127^2}$

      $$ s(w + 2p) \times s(h + 2p) = A $$ (10)

      正样本定义为响应得分图中的元素$u$在响应得分图中心$c$的搜索区域半径$R$内,其余的视作负样本,$k$为全卷积网络的总步长。文中设定$R = 2$$k = 1$

      $$ \left\{ {\begin{aligned} & { + 1}\quad {{\rm if} \;k\left\| {u - c} \right\| \leqslant R} \\ & { - 1} \quad {\rm otherwise} \end{aligned}} \right. $$ (11)

      (2) 网络模型的搭建

      按照表1定义的网络结构进行网络模型的搭建,按照VGG-Net-16预训练模型进行初始化。

      (3) 损失函数与优化器的选择

      损失函数采用逻辑回归损失函数,如公式(12)所示:

      $$ \ell \left( {y,v} \right) ={\rm log}\left( {1 + {\rm exp}\left( { - yv} \right)} \right) $$ (12)

      式中:$y \in \left\{ { + 1, - 1} \right\}$代表候选样本对对应的标签值;$v$代表单个图像样本对的实际相似计算得分值。整体响应图的损失函数定义为全部点的损失函数的均值,可表示为:

      $$ L\left( {y,v} \right) = \dfrac{1}{{\left| D \right|}}\sum\limits_{u \in D} {\ell \left( {y\left[ u \right],v\left[ u \right]} \right)} $$ (13)

      式中:$D$为响应得分图中像素数目;$u \in D$为响应得分图各个位置;$v\left[ u \right]$为每个位置的互相关卷积计算值各个搜索位置$u$的响应得分;$y\left[ u \right]$代表响应得分图各个位置的真值标签正负样本的划分。

      最小化目标函数如公式(14)所示,模型参数$\theta $指代的是网络的各层参数,采用随机梯度下降(SGD)对网络参数进行优化,从而得到网络参数的最优值,使其具有识别输入样本对特征共通性的能力。

      $$ \arg \mathop {\min }\limits_\theta = \mathop E\limits_{{\textit{z}},x,y} L\left( {y,f({\textit{z}},x;\theta )} \right) $$ (14)

      (4)网络训练

      训练时学习率设置为10−3~10−6。整个训练过程包含100多个阶段,每个阶段由6000对样本组成。每次计算8对样本的平均损失值。

    • 为分析文中通道-互联-空间注意力模块CISAM的作用,利用类激活热力图Grad-CAM[18]对不同的网络进行可视化,如图5所示。越敏感的位置温度越高,越不敏感的位置温度越低。第一排是未加CISAM的网络模型可视化结果,第二排是加入CISAM后的网络模型可视化结果。实验结果可以看出添加CISAM模块的网络的注意范围更广,能够更好地覆盖到所要识别的物体。当周围有相似物出现时,能更好地进行区分,将注意力集中在目标上,并未受到相似物体的干扰。

      图  5  Grad-CAM网络可视化结果

      Figure 5.  Grad-CAM network visualization results

    • 文中所提跟踪算法的实验平台硬件配置如表2所示。

      表 2  实验平台参数

      Table 2.  Experimental platform parameters

      DeviceProduct modelMemory
      CPUIntel(R)Core(TM)i7-970016G
      Basic frequency 3.0 GHz
      GPUNVIDIA GeForce RTX-20606G

      算法基于Pytorch实现,选择当前具有代表性的六种算法,包括文中所提算法、ECO[3]、SiamFC[5]、CFNet[6]、SiamRPN[7]、DasiamRPN[8],在目标跟踪通用数据集OTB-2015[19](涵盖11个属性,共100组视频)上测试评估算法的性能。以一次通过评估OPE (One Pass Evaluation)作为跟踪算法准确性能的评价标准。

    • 六种算法在数据集部分视频序列中的跟踪结果,如图6所示。不同算法的目标框使用不同的颜色表示,其中文中算法用红色表示。

      图  6  六种跟踪算法在部分视频序列上的定性结果显示

      Figure 6.  Qualitative results display of the 6 tracking algorithms over partial video sequences

      图6(a) CarDark序列中,目标时而变模糊,时而被遮挡,周围存在相似类干扰,同时光照情况也不断发生变化,文中算法由于可以筛选出适应性更强的目标特征,区分背景,所以始终未发生漂移。

      图6(b) MotorRolling序列中,目标自身发生超过360°的旋转,对特征表达能力提出了极大的挑战,除文中所提算法、SiamRPN以及DasiamRPN,其他算法在跟踪初始已经偏离了目标。

      图6(c) Skating1序列中,光照不断发生变化,且目标周围不断出现相似类干扰,多数算法已经远离目标,只有文中算法、SiamRPN以及ECO准确定位。

      图6(d) Skiing序列中,目标尺寸较小且速度较快,这对特征提取提出了更严格的要求,特征提取能力或者匹配能力低的算法在跟踪初始已经无法定位目标。

      图6(e) Suv序列中,目标清晰度低,偶尔超出视野范围,文中算法表现稳健,能够保持跟住目标。

      图6(f) Walking2序列中,目标尺度变化,中间偶有被遮挡,且有相似类干扰出现,文中算法能够避免遮挡出现时,跟踪漂移。

      通过对以上涵盖属性较为丰富的视频序列进行分析,文中所提算法在光照变化、尺度变化、旋转、相似物干扰、快速运动、分辨率低、超出视野等情况均有极佳的表现。

    • 跟踪算法主要通过对中心位置误差和覆盖率两个指标进行评估。前者是指跟踪结果与真实目标的中心位置之间的欧式距离;后者是指跟踪结果与真实目标的重叠率,均通过一定的阈值判定跟踪是否成功。二者分别体现在精度图以及成功率图中。

      (1)六种算法在OTB-2015上的跟踪精度曲线以及成功率曲线分别如图7图8所示。文中算法跟踪精度达到0.896,成功率为0.689,相对于SiamFC算法分别提高了12.4%、10.3%。证明文中算法在Siamese网络框架的基础上,提取了更深的网络特征,同时融入了注意力模块,使得网络提取到适应能力更强的特征,提升了算法的总体精度与稳健性。

      图  7  基于OTB-2015 OPE的跟踪精度曲线图

      Figure 7.  Tracking precision plots of OPE on OTB-2015 dataset

      (2)六种算法在OTB-2015中,11种不同属性视频的跟踪精度定量分析结果如图9所示,文中算法在尺度变化、形变、平面内旋转属性中排名第一,在光照变化、平面外旋转、遮挡、运动模糊、快速运动、超出视野、低分辨率属性中排名第二,仅次于ECO。

      图  8  基于OTB-2015 OPE的成功率曲线图

      Figure 8.  Success plots of OPE on OTB-2015 dataset

      图  9  OTB-2015 11种不同属性视频序列跟踪精度曲线

      Figure 9.  Tracking precision plots of 11 different attributes video sequences on OTB-2015 dataset

      (3)六种算法在OTB-2015中不同视频属性的成功率定量分析结果如图10所示,文中算法在尺度变化、形变、平面内旋转、低分辨率属性中排名第一,在光照变化、平面外旋转、遮挡、运动模糊、快速运动、超出视野、背景复杂属性中排名第二,仅次于ECO。

      图  10  OTB-2015 11种不同属性视频序列成功率曲线

      Figure 10.  Success plots of 11 different attributes video sequences on OTB-2015 dataset

      (4)相较于ECO,文中算法跟踪精度在部分场景仍有差距。然而基于深度网络特征的ECO算法在GPU上的速度为8 FPS[3],文中算法的平均速度达到37 FPS,在跟踪速度上有了提升。

      综上,文中所提算法在OTB-2015共11种属性的视频序列中表现稳定,通过融合注意力模块,嵌入更深的网络,取得了很好的跟踪效果,在提高跟踪精度与稳健性的同时,还保证了实时的跟踪速度,较好地适应跟踪场景的变化。

    • 文中对基于Siamese网络的目标跟踪算法进行了研究,提出一种嵌入深度卷积网络VGG-Net-16作为特征提取主干网络、融合注意力机制的端到端跟踪算法,并通过实验对其进行了验证,得到以下结论:

      (1)通道-互联-空间注意力模块的融入,显著提升了网络模型的特征提取能力、适应能力与判别能力。可视化分析实验表明,注意力模块的存在使得网络覆盖到目标更多的部位,学习更具代表性的特征,同时还可以排除周围相似物的干扰,区分性更强。

      (2)在通用目标跟踪数据集OTB-2015上的实验表明,与当前主流算法相比,文中算法能够达到较高的跟踪精度,在目标外观变化、相似物干扰、目标遮挡等复杂场景下,有稳健的表现。

      (3)在NVIDIA RTX 2060 GPU下的平均跟踪速度可达到37 FPS,满足实时应用的要求。

      注意力机制与目标跟踪算法的融合还处于探索阶段,后续工作会集中于注意力模块在网络中嵌套方式的研究,更进一步提升算法的性能。

参考文献 (19)

目录

    /

    返回文章
    返回