发布1万张水下物体实例分割数据集,助力AUV实时检测
The Common Objects Underwater (COU) Dataset for Robust Underwater Object Detection
- 构建包含24类常见水下人造物的实例分割数据集
- 训练模型在水下检测任务中性能显著优于仅用陆地数据训练的模型
- 覆盖池塘、湖泊、海洋等多环境,适合轻量级AUV检测系统研发
我们提出COU:Common Objects Underwater,一个包含约1万张实例分割图像的数据集,涵盖多种水下及海洋环境中常见的非生物人造物。数据来自多个水下机器人实地试验,覆盖封闭水域(泳池)和开放水域(湖泊、海洋)。现有水下数据集多聚焦于海洋生物,而COU填补了人造物类别多样性不足的空白,包含海洋垃圾、潜水工具、AUV等24类目标。为评估其有效性,我们采用三种前沿模型进行测试,结合标准精度与效率指标。结果显示,使用COU训练的检测器在性能上明显优于仅基于陆地数据训练的模型。COU已开源,供广泛使用。
原文摘要 · Abstract (English)
We introduce COU: Common Objects Underwater, an instance-segmented image dataset of commonly found man-made objects in multiple aquatic and marine environments. COU contains approximately 10K segmented images, annotated from images collected during a number of underwater robot field trials in diverse locations. COU has been created to address the lack of datasets with robust class coverage curated for underwater instance segmentation, which is particularly useful for training light-weight, real-time capable detectors for Autonomous Underwater Vehicles (AUVs). In addition, COU addresses the lack of diversity in object classes since the commonly available underwater image datasets focus only on marine life. Currently, COU contains images from both closed-water (pool) and open-water (lakes and oceans) environments, of 24 different classes of objects including marine debris, dive tools, and AUVs. To assess the efficacy of COU in training underwater object detectors, we use three state-of-the-art models to evaluate its performance and accuracy, using a combination of standard accuracy and efficiency metrics. The improved performance of COU-trained detectors over those solely trained on terrestrial data demonstrates the clear advantage of training with annotated underwater images. We make COU available for broad use under open-source licenses.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。