作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2021, Vol. 47 ›› Issue (2): 84-89. doi: 10.19678/j.issn.1000-3428.0057076

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

一种低参数的孪生卷积网络实时目标跟踪算法

罗朔, 侯进, 谭光鸿, 韩雁鹏   

  1. 西南交通大学 信息科学与技术学院, 成都 611756
  • 收稿日期:2019-12-31 修回日期:2020-02-05 出版日期:2021-02-15 发布日期:2020-02-09
  • 作者简介:罗朔(1994-),男,硕士研究生,主研方向为计算机视觉、深度学习;侯进(通信作者),副教授;谭光鸿、韩雁鹏,硕士研究生。
  • 基金资助:
    浙江大学CAD&CG国家重点实验室开放课题“基于深度学习的调制识别技术的研究”(A1823)。

A Low Parameter Real-Time Target Tracking Algorithm Based on Siamese Convolutional Network

LUO Shuo, HOU Jin, TAN Guanghong, HAN Yanpeng   

  1. The School of Information Science and Technology, Southwest Jiaotong University, Chengdu 611756, China
  • Received:2019-12-31 Revised:2020-02-05 Online:2021-02-15 Published:2020-02-09

摘要: 针对基于深度学习的目标跟踪算法模型参数多、难以部署于嵌入式设备上的问题,提出一种改进的孪生卷积网络实时目标跟踪算法。设计一个非对称卷积模块来构建整个网络框架,通过非对称卷积模块的压缩层减少模型参数量,利用非对称层进行特征融合,以在保证精度的同时压缩模型大小。使用三元组损失函数代替逻辑损失函数进行模型训练,在输入不变的情况下提取表达性更强的深度特征,从而完成目标跟踪任务并提高模型的跟踪精度。在GOT-10K、OTB100和VOT2016基准上对算法性能进行测试,结果表明,该算法能够将模型大小降为3.8×106,且速度与精度均优于SiamFC、KCF和DAT等跟踪算法。

关键词: 目标跟踪, 低参数模型, 孪生卷积网络, 实时性, 非对称卷积, 三元组损失

Abstract: The existing target tracking algorithms based on deep learning face deployment problems on embedded devices due to their large number of parameters.To address the problems,this paper proposes a real-time target tracking algorithm based on low parameter siamese convolutional network.By designing an asymmetric convolution module to build the entire network framework,the compression layer of the asymmetric convolution module is used to reduce the number of parameters,and the asymmetric layer is used for feature fusion to reduce the size of the model while maintaining accuracy.At the same time,the triple loss function is used to replace the logical loss function for model training,and the more expressive depth features are extracted for target tracking with the same input,which improves the tracking accuracy of the model.The performance of the algorithm is tested on the GOT-10K,OTB100 and VOT2016 benchmarks.Experimental results show that the algorithm reduces the model size to 3.8×106,and outperforms SiamFC,KCF,DAT and other tracking algorithms in terms of speed and accuracy.

Key words: target tracking, low parameter model, siamese convolutional network, real-time performance, asymmetric convolution, triplet loss

中图分类号: