Abstract:The random dumping of garbage in rivers has led to the continuous deterioration of water quality and affected people's living environment. The accuracy of detection of garbage floating in rivers is greatly affected by factors such as floating speed, night/daytime natural light, viewing angle and position, etc. This paper proposes a novel detection model, called YOLOv5_CBS, for the detection of garbage objects floating in rivers, based on improvements of the YOLOv5 model. Firstly, a coordinate attention (CA) mechanism is added to the original C3 module (without compressing the number of channels in the bottleneck), forming a new C3-CA-Uncompress Bottleneck (CCUB) module for improving the size of the receptive field and allowing the model to pay more attention to important parts of the processed images. Then, the Path Aggregation Network (PAN) in YOLOv5 is replaced with a Bidirectional Feature Pyramid Network (BiFPN), as proposed by other researchers, to enhance the depth of information mining and improve the feature extraction capability and detection performance of the model. In addition, the Complete Intersection over Union (CIoU) loss function, which was originally used in YOLOv5 for the calculation of location score of the compound loss, is replaced with the SCYLLA-IoU (SIoU) loss function, so as to speed up the model convergence and improve its regression precision. The results, obtained through experiments conducted on two datasets, demonstrate that the proposed YOLOv5_CBS model outperforms the original YOLOv5 model, along with three other state-of-the-art models (Faster R-CNN, YOLOv3, and YOLOv4), when used for river floating garbage objects detection, in terms of the recall, average precision, and F1 score achieved by reaching respective values of 0.885, 90.85%, and 0.8669 on the private dataset, and 0.865, 92.18%, and 0.9006 on the Flow-Img public dataset.

GST-YOLO: a lightweight visual detection algorithm for underwater garbage detection

Water surface garbage detection based on lightweight YOLOv5

YOLOv8-C2f-Faster-EMA: An Improved Underwater Trash Detection Model Based on YOLOv8

A modified YOLOv4 detection method for a vision-based underwater garbage cleaning robot

YOLOv7t-CEBC Network for Underwater Litter Detection

A modified YOLOv3 detection method for vision-based water surface garbage capture robot

Underwater Object Detection Based on Enhanced YOLO

A Lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU on YOLOv8

Underwater target detection and embedded deployment based on lightweight YOLO_GN

Detection of River Floating Garbage Based on Improved YOLOv5

Lightweight object detection algorithm based on YOLOv5 for unmanned surface vehicles

Attention-Based Lightweight YOLOv8 Underwater Target Recognition Algorithm

An automatic garbage detection using optimized YOLO model

Underwater Object Detection Algorithm Based on an Improved YOLOv8

A novel finetuned YOLOv8 model for real-time underwater trash detection

Research on the Identification and Classification of Marine Debris Based on Improved YOLOv8

YOLO‐UOD: An underwater small object detector via improved efficient layer aggregation network

APM-YOLOv7 for Small-Target Water-Floating Garbage Detection Based on Multi-Scale Feature Adaptive Weighted Fusion

Underwater Robot Target Detection Algorithm Based on YOLOv8

EcoDetect-YOLO: A Lightweight, High-Generalization Methodology for Real-Time Detection of Domestic Waste Exposure in Intricate Environmental Landscapes

YOLOv9-YX: lightweight algorithm for underwater target detection