【技术实现步骤摘要】
一种基于深度学习的高密度行人头部实时跟踪算法
[0001]本专利技术属于高密度行人重识别领域,针对拥挤场景下的行人头部跟踪,具体设计成基于EPSA
‑
CenterNet和UKF的多目标跟踪算法模型。
技术介绍
[0002]近些年来,深度学习技术迅速发展,在行人重识别领域中得到广泛应用。但由于我国人口基数过大,在多目标全身跟踪中经常发生遮挡情况。将深度学习应用到拥挤场景下的全身跟踪普遍跟踪精度不高,ID切换频繁,严重影响到公共场合中对行人轨迹的跟踪,不利于社会治安的维护和管理。并且目前现有算法对行人头部这种小目标跟踪仍然常发生漏检和误检等情况,加上遮挡以及运动变化等复杂情况会影响对目标的身份判定。
[0003]目前基于深度学习的多目标跟踪方法主要通过目标检测和数据关联的方法来实现,主要有以下几种:
[0004]1.基于anchor
‑
based的目标检测算法
[0005]2.基于DeepSort算法和贪婪匹配策略的跟踪匹配
[0006]以上方法能在一定程度上提 ...
【技术保护点】
【技术特征摘要】
1.一种基于深度学习的高密度行人头部实时跟踪算法,其特征在于,结合了EPSA
‑
CenterNet检测器与基于UKF
‑
DeepSort的跟踪器。针对拥挤场景下行人头部小目标特征难以检测问题,提出了多尺度提取SPC模块并引入注意力机制来增强小目标特征检测精度。针对拥挤场景下行人遮挡问题,提出了UKF
‑
DeepSort算法运用sigma点集模拟行人运动的分布,并结合CS
‑
Softmax优化网络中ReID分支中分类损失函数以增强模型重识别能力。算法主要有以下步骤:步骤一:在EPSA
‑
CenterNet的检测算法上作出改进,得到目标检测网络,并训练成行人头部的检测模型。步骤二:在特征提取网络输出端的ReID分支应用CS
‑
Softmax损失函数以提高头部分类效果。步骤三:将目标的特征信息输入到改进后的跟踪器中进行数据关联匹配,运用UKF滤波器中的sigma点集模拟真实状态并更新系统状态,完成多目标跟踪模型的搭建。2.根据权利要求1所述,EPSA
‑
CenterNet检测算法是以原CenterNet网络结构所改进的,其特征在于,主干网络和特征提取网络以及四条输出头,包括:主干网络是以ResNet
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。