首个针对破浪流实例分割的基准数据集与YOLOv8基线模型。
Rip Current Segmentation: A Novel Benchmark and YOLOv8 Baseline Results
- 构建2466张图像+17段无人机视频的标注数据集,支持实例分割训练与测试。
- YOLOv8-nano在验证集上达88.94% mAP50,测试集宏平均81.21%。
- 代码与数据公开,适合海洋安全、计算机视觉研究者参考。
破浪流是全球许多海滩致命事故和伤害的主要原因,因此自动检测这些危险表层水流至关重要。本文提出一项新任务:破浪流实例分割。我们构建了一个包含2,466张图像的综合数据集,其中新增了用于实例分割的多边形标注,用于训练与验证。此外,还提出一个由17段无人机视频组成的新型数据集(约24,000帧,30 FPS),标注了多边形(实例分割)与边界框(目标检测),用于测试。我们在静态图像上训练多种YOLOv8版本进行实例分割,并评估其在测试数据集上的表现。最佳结果由YOLOv8-nano模型达成(可在便携设备运行),在验证集上mAP50为88.94%,测试集宏平均为81.21%。该结果为未来破浪流分割研究提供了基准。本工作通过引入详细标注数据集并训练深度学习模型,推动了该领域的进展。代码、训练细节及标注数据集已公开于https://github.com/Irikos/rip_currents。
原文摘要 · Abstract (English)
Rip currents are the leading cause of fatal accidents and injuries on many beaches worldwide, emphasizing the importance of automatically detecting these hazardous surface water currents. In this paper, we address a novel task: rip current instance segmentation. We introduce a comprehensive dataset containing $2,466$ images with newly created polygonal annotations for instance segmentation, used for training and validation. Additionally, we present a novel dataset comprising $17$ drone videos (comprising about $24K$ frames) captured at $30 FPS$, annotated with both polygons for instance segmentation and bounding boxes for object detection, employed for testing purposes. We train various versions of YOLOv8 for instance segmentation on static images and assess their performance on the test dataset (videos). The best results were achieved by the YOLOv8-nano model (runnable on a portable device), with an mAP50 of $88.94%$ on the validation dataset and $81.21%$ macro average on the test dataset. The results provide a baseline for future research in rip current segmentation. Our work contributes to the existing literature by introducing a detailed, annotated dataset, and training a deep learning model for instance segmentation of rip currents. The code, training details and the annotated dataset are made publicly available at https://github.com/Irikos/rip_currents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。