3 个月前

在无人飞行器目标检测中利用合成数据

在无人飞行器目标检测中利用合成数据

摘要

在无人飞行器(UAV)上训练基于深度学习的目标检测器,通常需要大量真实数据,而获取这些数据不仅成本高昂、耗时较长,且在某些特定环境中甚至可能违反法律法规。相比之下,合成数据的获取则快速且成本低廉。本文探讨了在多种应用场景下,利用合成数据进行UAV目标检测的潜力。为此,我们对开源框架DeepGTAV进行了扩展,使其适用于UAV场景。我们在多个领域内生成了多组大规模、高分辨率的合成数据集,并通过分析多种模型在不同训练策略下的表现,验证了这些数据在真实世界UAV目标检测任务中的有效性。此外,本文还系统研究了多种数据生成与采样参数的影响,为后续科学研究提供了可操作的工程指导建议。DeepGTAV框架的源代码已开源,获取地址为:https://git.io/Jyf5j。

代码仓库

Eisbaer8/DeepGTAV
官方
GitHub 中提及

基准测试

基准方法指标
object-detection-on-seadronesseeSynth Pretrained Faster R-CNN ResNeXt-101-FPN
mAP@0.5: 59.20
object-detection-on-seadronesseeSynth Pretrained EffDetD0
mAP@0.5: 38.74
object-detection-on-seadronesseeYolo 5
mAP@0.50: 54.74
object-detection-on-seadronesseeSynth Pretrained Yolo5
mAP@0.5: 59.08

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
在无人飞行器目标检测中利用合成数据 | 论文 | HyperAI超神经