构建10万张海图数据集,专为复杂海况下小船检测设计。
WUTDet: A 100K-Scale Ship Detection Dataset and Benchmarks with Dense Small Objects
- 构建包含10万图像、38万目标的大型船舶检测数据集
- Transformer模型在小目标检测上表现最佳,精度领先12%
- 适合做海事智能感知算法评估与泛化能力研究
船舶检测是智能水路交通系统中的基础感知任务。然而,现有公开数据集在规模、小目标占比和场景多样性方面仍显不足,限制了检测算法在复杂海况下的系统性评估与泛化研究。为此,我们构建了大规模船舶检测数据集WUTDet,包含100,576张图像和381,378个标注的船舶实例,覆盖港口、锚地、航行、靠泊等多种操作场景,以及雾、强光、低光照、降雨等多样成像条件,具有显著的多样性与挑战性。基于WUTDet,我们系统评估了来自三种主流架构(CNN、Transformer、Mamba)的20个基准模型。实验表明,Transformer架构在整体检测精度(AP)和小目标检测性能(APs)上表现最优,适应复杂海况能力更强;CNN架构在推理效率上保持优势,更适合实时应用;而Mamba架构在精度与计算效率间取得良好平衡。此外,我们构建了统一的跨数据集测试集Ship-GEN,用于评估模型泛化能力。在Ship-GEN上的结果表明,基于WUTDet训练的模型在不同数据分布下展现出更强的泛化能力。这些发现证明WUTDet为复杂海况下船舶检测算法的研究、评估与泛化分析提供了有效数据支持。数据集已公开:https://github.com/MAPGroup/WUTDet。
原文摘要 · Abstract (English)
Ship detection for navigation is a fundamental perception task in intelligent waterway transportation systems. However, existing public ship detection datasets remain limited in terms of scale, the proportion of small-object instances, and scene diversity, which hinders the systematic evaluation and generalization study of detection algorithms in complex maritime environments. To this end, we construct WUTDet, a large-scale ship detection dataset. WUTDet contains 100,576 images and 381,378 annotated ship instances, covering diverse operational scenarios such as ports, anchorages, navigation, and berthing, as well as various imaging conditions including fog, glare, low-lightness, and rain, thereby exhibiting substantial diversity and challenge. Based on WUTDet, we systematically evaluate 20 baseline models from three mainstream detection architectures, namely CNN, Transformer, and Mamba. Experimental results show that the Transformer architecture achieves superior overall detection accuracy (AP) and small-object detection performance (APs), demonstrating stronger adaptability to complex maritime scenes; the CNN architecture maintains an advantage in inference efficiency, making it more suitable for real-time applications; and the Mamba architecture achieves a favorable balance between detection accuracy and computational efficiency. Furthermore, we construct a unified cross-dataset test set, Ship-GEN, to evaluate model generalization. Results on Ship-GEN show that models trained on WUTDet exhibit stronger generalization under different data distributions. These findings demonstrate that WUTDet provides effective data support for the research, evaluation, and generalization analysis of ship detection algorithms in complex maritime scenarios. The dataset is publicly available at: https://github.com/MAPGroup/WUTDet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。