关键词:
目标跟踪
目标检测
嵌入式系统
YOLO v4
DeepSORT
摘要:
以目标跟踪为基础服务的智能交通对人们的生活起着越来越重要的作用,大量目标跟踪系统正逐步被部署到日常交通中。综合考虑成本、跟踪能力等因素,现有目标跟踪系统更多采用客户端-服务器-客户端的部署方式,嵌入式终端只进行数据的采集和结果的呈现,云端服务器则进行目标跟踪结果的解算。此种耦合度较大的系统无法适应实验敏感型业务的需求。因此,本文面向嵌入式终端,研究设计了一种基于Deep SORT的轻量级目标跟踪算法,并实现了一个实时的嵌入式目标跟踪系统。该系统脱离了云服务器和网络的束缚,使得所有流程都在嵌入式侧进行,提升了目标跟踪系统的实时性和效率。直接部署目标跟踪算法对性能较差的嵌入式终端是一个巨大的考验,因此本文的核心目标是以Deep SORT算法为主要跟踪手段,对该算法进行轻量化。首先,对Deep SORT算法的基础目标检测器进行优化,提出了基于YOLO v4的改进目标检测算法。参考了 MobileNet基本模块的结构,设计了一个轻量级网络,对原始YOLO v4的主干网络进行替换。同时,利用深度可分离卷积的思想对特征融合网络中的普通卷积进行优化,利用通道剪枝技术对整体网络进一步瘦身。最终,基于YOLO v4的改进算法在检测精度相当甚至略有提升的同时,检测速度得到了大幅度的提高。然后,在基础检测器轻量化的基础之上,本文还对Deep SORT算法流程进行了改进。第一步利用扩展卡尔曼滤波算法对检测框的预测过程进行改进,使得预测非线性场景下的检测框位置更加准确。第二步,使用广义交并比对算法流程中使用到的普通交并比度量方式进行替换,使得两个目标的相对位置衡量更加精准。紧接着,考虑到真实交通中行人和车辆的运动状态,本文提出了一种基于运动判断的自适应跟踪算法,结合目标的真实运动状态,改进了 Deep SORT的流程,减小了目标检测算法的调用频率,提升了跟踪算法的效率和速度。最后,本文以上述改进Deep SORT目标跟踪算法为基础,设计并实现了一个实时的嵌入式目标跟踪系统,该系统只由一个嵌入式终端组成,不需要服务器和网络的参与,使得对目标的信息采集、数据处理、跟踪以及后处理都在嵌入式终端中形成闭环。最终,通过实地测试表明,该系统的跟踪能力和处理速度在真实交通中都有着不错的表现。