Wang Yuhang, Ye Hua, Shu Xin
School of Computer Science, Jiangsu University of Science and Technology, Zhenjiang 212100, China.
Sensors (Basel). 2025 Jun 22;25(13):3890. doi: 10.3390/s25133890.
Underwater images often exhibit characteristics such as low contrast, blurred and small targets, object clustering, and considerable variations in object morphology. Traditional detection methods tend to be susceptible to omission and false positives under these circumstances. Furthermore, owing to the constrained memory and limited computing power of underwater robots, there is a significant demand for lightweight models in underwater object detection tasks. Therefore, we propose an enhanced lightweight YOLOv10n-based model, BSE-YOLO. Firstly, we replace the original neck with an improved Bidirectional Feature Pyramid Network (Bi-FPN) to reduce parameters. Secondly, we propose a Multi-Scale Attention Synergy Module (MASM) to enhance the model's perception of difficult features and make it focus on the important regions. Finally, we integrate Efficient Multi-Scale Attention (EMA) into the backbone and neck to improve feature extraction and fusion. The experiment results demonstrate that the proposed BSE-YOLO reaches 83.7% @0.5 on URPC2020 and 83.9% @0.5 on DUO, with the parameters reducing 2.47 M. Compared to the baseline model YOLOv10n, our BSE-YOLO improves @0.5 by 2.2% and 3.0%, respectively, while reducing the number of parameters by approximately 0.2 M. The BSE-YOLO achieves a good balance between accuracy and lightweight, providing an effective solution for underwater object detection.
水下图像通常具有对比度低、目标模糊且小、目标聚类以及目标形态变化大等特点。在这些情况下,传统检测方法往往容易出现漏检和误报。此外,由于水下机器人的内存受限和计算能力有限,水下目标检测任务对轻量级模型有很大需求。因此,我们提出了一种基于YOLOv10n的增强型轻量级模型BSE - YOLO。首先,我们用改进的双向特征金字塔网络(Bi - FPN)替换原始的颈部,以减少参数。其次,我们提出了一种多尺度注意力协同模块(MASM),以增强模型对困难特征的感知,并使其专注于重要区域。最后,我们将高效多尺度注意力(EMA)集成到主干和颈部,以改进特征提取和融合。实验结果表明,所提出的BSE - YOLO在URPC2020上@0.5时达到83.7%,在DUO上@0.5时达到83.9%,参数减少了2.47M。与基线模型YOLOv10n相比,我们的BSE - YOLO在@0.5时分别提高了2.2%和3.0%,同时参数数量减少了约0.2M。BSE - YOLO在准确性和轻量级之间实现了良好的平衡,为水下目标检测提供了有效的解决方案。