An Efficient YOLO Network With CSPCBAM, Ghost, and Cluster-NMS for Underwater Target Detection

作者：Zheng Zhang, Qingshan Tong, Xiaofei Huang · 发表于：IEEE Access · 年份：2024 · DOI：10.1109/access.2024.3368878 · 被引用次数：16 · 研究领域：Underwater Acoustics Research、Underwater Vehicles and Communication Systems、Water Quality Monitoring Technologies

In recent years, owing to the rapid advancements in deep learning, advanced object detection methods, such as You Only Look Once (YOLO) and Efficient Detector (EfficientDet), have been frequently used to detect underwater organisms. However, due to the complexity of underwater scenarios and deployment limitations, these models often encounter various challenges, such as blurred targets, occlusions, and high model computing costs. On this basis, we propose a YOLO network (CGC-YOLO) based on Cross-Stage Partial Convolutional Block Attention Module (CSPCBAM), Ghost module, and cluster non-maximum suppression (Cluster-NMS). Firstly, CSPCBAM enhances the model’s ability to extract intricate features by amplifying pertinent feature information across both channel and spatial dimensions. This augmentation contributes to an improved detection performance of the model, especially when dealing with fuzzy targets. Secondly, the Ghost module is employed to optimize the model’s efficiency by decreasing its parameters and reducing the computational load in terms of floating-point operations per second (FLOPs). Finally, by introducing Cluster-NMS and Score Penalty Mechanism (SPM) to reweight the confidence of bounding boxes, the model can retain the real object with occlusion. The experimental results show that on the Underwater Robot Picking Competition 2020 (URPC 2020) and brackish water dataset, the mAP@0.5 of our proposed CGC-YOLO reaches 87.2% and 98.6% respectively, which is at least ...