Department of Cyberspace Security, Hainan University, Haikou, Hainan Province, China.
PLoS One. 2024 Sep 27;19(9):e0311173. doi: 10.1371/journal.pone.0311173. eCollection 2024.
Underwater object detection plays a crucial role in safeguarding and exploiting marine resources effectively. Addressing the prevalent issues of limited storage capacity and inadequate computational power in underwater robots, this study proposes FEB-YOLOv8, a novel lightweight detection model. FEB-YOLOv8, rooted in the YOLOv8 framework, enhances the backbone network by refining the C2f module and introducing the innovative P-C2f module as a replacement. To compensate for any potential reduction in detection accuracy resulting from these modifications, the EMA module is incorporated. This module augments the network's focus on multi-scale information, thus boosting its feature extraction capabilities. Furthermore, inspired by Bi-FPN concepts, a new feature pyramid network structure is devised, achieving an optimal balance between model lightness and detection precision. The experimental results on the underwater datasets DUO and URPC2020 reveal that our FEB-YOLOv8 model enhances the mAP by 1.2% and 1.3% compared to the baseline model, respectively. Moreover, the model's GFLOPs and parameters are lowered to 6.2G and 1.64M, respectively, marking a 24.39% and 45.51% decrease from the baseline model. These experiments validate that FEB-YOLOv8, by harmonizing lightness with accuracy, presents an advantageous solution for underwater object detection tasks.
水下目标检测在有效保护和利用海洋资源方面起着至关重要的作用。针对水下机器人存储容量有限和计算能力不足的普遍问题,本研究提出了 FEB-YOLOv8,这是一种新颖的轻量级检测模型。FEB-YOLOv8 基于 YOLOv8 框架,通过改进 C2f 模块和引入创新的 P-C2f 模块来增强骨干网络。为了弥补这些修改可能导致的检测精度下降,引入了 EMA 模块。该模块增强了网络对多尺度信息的关注,从而提高了其特征提取能力。此外,受 Bi-FPN 概念的启发,设计了一种新的特征金字塔网络结构,在模型轻量化和检测精度之间实现了最佳平衡。在水下数据集 DUO 和 URPC2020 上的实验结果表明,与基线模型相比,我们的 FEB-YOLOv8 模型分别将 mAP 提高了 1.2%和 1.3%。此外,模型的 GFLOPs 和参数分别降低到 6.2G 和 1.64M,与基线模型相比分别降低了 24.39%和 45.51%。这些实验验证了 FEB-YOLOv8 通过协调轻量化和准确性,为水下目标检测任务提供了有利的解决方案。