A lightweight model for efficient detection of small floating debris in complex aquatic environments
摘要
Detection and removal of floating objects are crucial for water pollution control and the development of sustainable aquatic ecosystems. While unmanned platforms offer a promising solution, their limited computational resources, coupled with complex background interference (e.g., illumination, reflection), make accurate and efficient detection of small objects a major challenge. To overcome this challenge, we propose the SG-YOLO model, a novel lightweight model that excels in complex scenes and small object detection. Our approach incorporates three key innovations to address the core challenges: A lightweight cross-channel architecture combining heavily parameterized convolutions and an attention mechanism to enhance feature representation and robustness to complex interferences. The introduction of SPD-Conv and a specially designed SG-C2f module to comprehensively preserve and refine the feature details of small objects and prevent them from being lost in deep networks. A detection head with multi-scale feature fusion to enhance scale invariance and improve the accuracy and efficiency of small object detection. Extensive experiments on the IWHR_AI_Lable_Floater_V1 and FLOW-IMG datasets demonstrate that SG-YOLO achieves mAP@0.5 accuracy of 91.8 and 84.2%, respectively, improving upon YOLOv8 by 2.4 and 5.0%, while maintaining a parameter size of only 3.3 M and achieving a frame rate of 131. Field tests and quantitative environmental performance analysis further validate its practicality, highlighting SG-YOLO’s excellent balance between lightweight design and efficient small object detection.