arXiv:2504.15694cs.CV2025-04被引 9

轻量实时水下目标检测,参数仅120万却达顶尖精度。

You Sense Only Once Beneath: Ultra-Light Real-Time Underwater Object Detection

  • 用多谱小波编码减少水下色偏导致的语义损失。
  • 动态选择关键信息,模型仅120万参数仍达83.1% mAP50。
  • 适合边缘设备部署,推理速度超主流模型20%以上。

尽管目标检测已取得显著进展,但在低图像质量与计算资源受限的水下环境中,模型精度与效率仍有提升空间。为此,我们提出超轻量实时水下目标检测框架 YSOOB。通过多谱小波编码(MSWE)在频域对输入图像进行处理,有效缓解水下光学色偏带来的语义损失;重新审视偶数尺寸与转置卷积特性,使模型在重采样过程中动态选择并增强关键信息,提升泛化能力;最后通过通道压缩与重构大核卷积(RLKC)消除冗余,实现模型轻量化。最终构建的 YSOOB 模型仅含 120 万参数,在 URPC2020 与 DUO 数据集上分别达到 83.1% 与 82.9% 的 mAP50,性能媲美当前最优模型。在 T4 GPU(TensorRT FP16)和 Jetson Xavier NX 边缘设备上,推理速度分别达 781.3 FPS 与 57.8 FPS,较 YOLOv12-N 提升 28.1% 和 22.5%。

原文摘要 · Abstract (English)

Despite the remarkable achievements in object detection, the model's accuracy and efficiency still require further improvement under challenging underwater conditions, such as low image quality and limited computational resources. To address this, we propose an Ultra-Light Real-Time Underwater Object Detection framework, You Sense Only Once Beneath (YSOOB). Specifically, we utilize a Multi-Spectrum Wavelet Encoder (MSWE) to perform frequency-domain encoding on the input image, minimizing the semantic loss caused by underwater optical color distortion. Furthermore, we revisit the unique characteristics of even-sized and transposed convolutions, allowing the model to dynamically select and enhance key information during the resampling process, thereby improving its generalization ability. Finally, we eliminate model redundancy through a simple yet effective channel compression and reconstructed large kernel convolution (RLKC) to achieve model lightweight. As a result, forms a high-performance underwater object detector YSOOB with only 1.2 million parameters. Extensive experimental results demonstrate that, with the fewest parameters, YSOOB achieves mAP50 of 83.1% and 82.9% on the URPC2020 and DUO datasets, respectively, comparable to the current SOTA detectors. The inference speed reaches 781.3 FPS and 57.8 FPS on the T4 GPU (TensorRT FP16) and the edge computing device Jetson Xavier NX (TensorRT FP16), surpassing YOLOv12-N by 28.1% and 22.5%, respectively.

目标检测轻量化水下视觉边缘计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。