首页 | 本学科首页   官方微博 | 高级检索  
     

基于YOLOv3和ASMS的目标跟踪算法
引用本文:吕晨,程德强,寇旗旗,庄焕东,李海翔.基于YOLOv3和ASMS的目标跟踪算法[J].光电工程,2021(2):67-77.
作者姓名:吕晨  程德强  寇旗旗  庄焕东  李海翔
作者单位:中国矿业大学信息与控制工程学院;中国矿业大学计算机科学与技术学院
基金项目:国家自然科学基金资助项目(51774281)。
摘    要:为了解决传统算法在全自动跟踪过程中遇到遮挡或运动速度过快时的目标丢失问题,本文提出一种基于YOLOv3和ASMS的目标跟踪算法。首先通过YOLOv3算法进行目标检测并确定跟踪的初始目标区域,然后基于ASMS算法进行跟踪,实时检测并判断目标跟踪效果,通过二次拟合定位和YOLOv3算法实现跟踪目标丢失后的重新定位。为了进一步提升算法运行效率,本文应用增量剪枝方法,对算法模型进行了压缩。通过与当前主流算法进行对比,实验结果表明,本算法能够很好地解决受到遮挡时跟踪目标的丢失问题,提高了目标检测和跟踪的精度,且具有计算复杂度低、耗时少,实时性高的优点。

关 键 词:目标跟踪  目标丢失  YOLOv3  模型剪枝  ASMS

Target tracking algorithm based on YOLOv3 and ASMS
Lv Chen,Cheng Deqiang,Kou Qiqi,Zhuang Huandong,Li Haixiang.Target tracking algorithm based on YOLOv3 and ASMS[J].Opto-Electronic Engineering,2021(2):67-77.
Authors:Lv Chen  Cheng Deqiang  Kou Qiqi  Zhuang Huandong  Li Haixiang
Affiliation:(School of Information and Control Engineering,China University of Mining and Technology,Xuzhou,Jiangsu 221000,China;School of Computer Science&Technology,China University of Mining and Technology,Xuzhou,Jiangsu 221000,China)
Abstract:In order to solve the problem of loss when the target encounters occlusion or the speed is too fast during the automatic tracking process,a target tracking algorithm based on YOLOv3 and ASMS is proposed.Firstly,the target is detected by the YOLOv3 algorithm and the initial target area to be tracked is determined.After that,the ASMS algorithm is used for tracking.The tracking effect of the target is detected and judged in real time.Repositioning is achieved by quadratic fitting positioning and the YOLOv3 algorithm when the target is lost.Finally,in order to further improve the efficiency of the algorithm,the incremental pruning method is used to compress the algorithm model.Compared with the mainstream algorithms,experimental results show that the proposed algorithm can solve the lost problem when the tracking target is occluded,improving the accuracy of target detection and tracking.It also has advantages of low computational complexity,time-consuming,and high real-time performance.
Keywords:target tracking  target loss  you look only once v3  model pruning  robust scale-adaptive mean-shift
本文献已被 维普 等数据库收录!
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号