首页 >

无人机状态检测Kalman滤波空地目标跟踪算法

徐心宇1,陈建2   

  1. 1. 中国农业大学工学院
    2. 中国农业大学
  • 收稿日期:2023-11-06 修回日期:2024-03-03 出版日期:2024-03-11 发布日期:2024-03-11
  • 通讯作者: 陈建
  • 基金资助:
    国家自然科学基金

UAV object tracking for air-ground targets based on status detection and Kalman filter

Xin-Yu XU1,Jian Chen   

  • Received:2023-11-06 Revised:2024-03-03 Online:2024-03-11 Published:2024-03-11
  • Contact: Jian Chen

摘要: 针对无人机(UAV)面向空地目标进行目标跟踪过程中,发生目标离开视野、目标被遮挡、存在相似目标干扰等问题导致追踪失败的情况,提出一种基于追踪状态检测和Kalman滤波的重定位更新机制,将其与孪生全卷积网络(siamfc)跟踪器结合。本文以空地目标为被跟踪对象,以无人机为跟踪空地目标的跟踪者。首先,基于双峰选择、平均峰值相关能量变化率、最高响应值变化率和峰值旁瓣比变化率的检测机制对当前的追踪状态是否异常进行检测,判断siamfc的追踪结果是否满足作为观测值的要求。其次,Kalman滤波利用目标运动的先验信息对追踪进行预测更新,当追踪状态异常时能够及时校正调整目标跟踪结果。基于LaSOT数据集完成训练,在UAV123航空数据集和自制的以无人机为目标的数据集上进行实时目标跟踪测试和对比实验。实验结果表明:该算法在UAV123上的精确率和成功率为66.0%和47.4%(62fps),在自制的以无人机为目标的数据集上的精确率和成功率为72.0%和58.6%(55fps),满足目标跟踪的实时性要求,且跟踪结果优于多数跟踪器。该算法在无人机为跟踪者和被跟踪对象的情况下均能完成有效目标跟踪,应对目标离开视野、部分或全部遮挡和存在相似目标干扰等挑战性场景的能力有所增强,且算法具有良好的泛化能力。

关键词: 无人机, 目标跟踪, 孪生网络, Kalman滤波, 重定位机制, 追踪状态检测

Abstract: Aiming at the target tracking process of unmanned aerial vehicle (UAV) for air-ground targets, which occurs when the target leaves the field of view, the target is occluded, and the existence of similar target interference leads to tracking failure, a relocation mechanism based on the tracking status detection and Kalman filter is proposed, which is combined with the fully convolutional Siamese network (siamfc) tracker. In this paper, an air-ground target is taken as the object to be tracked, and an unmanned aerial vehicle (UAV) is taken as the tracker to track the air-ground target. First, the detection mechanism based on the selection mechanism for double peaks, changing rate of average peak correlation energy, changing rate of maximum response value, and changing rate of peak to sidelobe ratio detects whether the current track-ing status is abnormal or not to determine whether the tracking results of the siamfc satisfy the requirement of being an observation. Secondly, Kalman filter utilizes a priori information of the target motion to predict and update the tracking result. When the tracking status is abnormal, it can correct and adjust the tracking result in time. We use the LaSOT video dataset to train the network. Real-time object tracking tests and comparison experiments were operated on UAV123 aerial dataset and the customized dataset with UAV as the target. The experimental results show that the algorithm has an accuracy and success rate of 66.0% and 47.4% (62fps) on UAV123, and 72.0% and 58.6% (55fps) on the customized dataset with UAV as the target, which meets the real-time requirements of UAV object tracking and the tracking results are better than most trackers. The algorithm accomplishes effective target tracking with both UAVs as trackers and tracked objects, and has an enhanced ability to cope with challenging scenarios such as the target leaving the field of view, partial or total occlusion, and similar target interference. Meanwhile, the algorithm has good generalization.

Key words: UAV, Object Tracking, Siamese Network, Kalman Filter, Relocation Mechanism, Tracking Status Detection

中图分类号: