3 个月前

轨迹增强与合成数据辅助的无人机检测

轨迹增强与合成数据辅助的无人机检测

摘要

这是由AVSS 2021主办的“无人机vs.鸟类挑战赛”(Drone vs. Bird Challenge)一等奖获奖方案所对应的论文。随着无人机成本降低与技术不断进步,无人机检测已成为一项关键的目标检测任务。然而,在对比度弱、远距离、能见度低等不利条件下实现远距离无人机检测,仍需依赖高效的算法。本文提出一种基于卡尔曼滤波(Kalman-based)目标跟踪器,结合真实数据与合成数据对YOLOv5模型进行微调的检测方法,以提升检测置信度。实验结果表明,通过引入最优子集的合成数据对真实数据进行增强,可有效提升模型性能;此外,利用目标跟踪方法获取的时序信息,还可进一步提升检测效果。

基准测试

基准方法指标
object-detection-on-drone-vs-birdOBSS YOLOv5+Track Boosting (Including Synthetic Data)
AP50: 79.4
AP50l: 70.3
AP50m: 72.7
AP50s: 86.2
object-detection-on-drone-vs-birdOBSS YOLOv5+Track Boosting
AP50: 76.1
AP50l: 43
AP50m: 67.6
AP50s: 86.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
轨迹增强与合成数据辅助的无人机检测 | 论文 | HyperAI超神经