3 个月前

SeaDronesSee:开放水域中检测人类的海洋基准数据集

SeaDronesSee:开放水域中检测人类的海洋基准数据集

摘要

无人飞行器(Unmanned Aerial Vehicles, UAVs)在海上搜救任务中具有至关重要的作用,因其具备灵活且快速的作业能力。现代计算机视觉算法在辅助此类任务方面受到广泛关注。然而,这些算法依赖于大量来自无人机的真实场景训练数据,而目前此类数据仅在陆地交通场景中较为丰富。此外,现有的目标检测与跟踪数据集普遍缺乏或仅提供有限的环境信息,忽视了极具价值的上下文线索。为此,本文提出一个大规模的视觉目标检测与跟踪基准数据集——SeaDronesSee,旨在弥合陆基视觉系统与海基系统之间的差距。本研究收集并标注了超过54,000帧图像,涵盖约40万个目标实例,数据采集自5至260米的多种飞行高度及0至90度的多样化视角,并同步提供了包括飞行高度、视角角度在内的详细元数据信息。我们基于该新建立的基准数据集,对多种前沿计算机视觉算法进行了评估,以提供可参考的基准性能。同时,我们搭建了在线评估服务器,研究人员可上传其预测结果,并在统一的中央排行榜上进行结果对比与性能排名。

基准测试

基准方法指标
multi-object-tracking-on-seadronesseeFairMOT R34
MOTA: 0.3050
multi-object-tracking-on-seadronesseeTracktor++
MOTA: 0.7190
multi-object-tracking-on-seadronesseeFairMOT D34
MOTA: 0.3650
object-detection-on-seadronesseeFaster R-CNN ResNeXt-101-FPN
mAP@0.5: 54.66
object-detection-on-seadronesseeCenterNet ResNet18
mAP@0.5: 21.84
object-detection-on-seadronesseeCenterNet Hourglass104
mAP@0.5: 50.32
object-detection-on-seadronesseeFaster RCNN ResNet50FPN
mAP@0.5: 30.09
object-detection-on-seadronesseeEfficientDet D0
mAP@0.5: 37.11
object-detection-on-seadronesseeCenterNet ResNet101
mAP@0.5: 36.42
object-tracking-on-seadronesseePrDiMP50
Precision Score: 84.93010
Success Rate: 67.00010
object-tracking-on-seadronesseeDiMP50
Precision Score: 86.84020
Success Rate: 67.33400
object-tracking-on-seadronesseeDiMP18
Precision Score: 82.70010
Success Rate: 64.60100
object-tracking-on-seadronesseeAtom
Precision Score: 82.32100
Success Rate: 63.80000
object-tracking-on-seadronesseePrDiMP18
Precision Score: 83.51010
Success Rate: 65.90210

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SeaDronesSee:开放水域中检测人类的海洋基准数据集 | 论文 | HyperAI超神经