YOLO-EcoFocus: small waste bottle detection in inland water bodies focusing on sustainable environmental cleaning using USVs
摘要
The alarming issue of plastic waste polluting natural inland aquatic environments necessitates the development of automated trash detection techniques that work effectively in complex and dynamic inland water environments. Several deep learning techniques have been developed for detecting floating trash but suffer due to objects’ tiny size, occlusions, complex backgrounds, etc. Hence, the paper proposes a novel waste detection technique, YOLO-Ecofocus, employing YOLOv9-gelan-c architecture, augmented with an efficient attention mechanism and parallel Shifted Window (Swin) transformer block for detecting small floating plastic waste bottles. The work derives the convolutional channel and spatial attention network (CSA-Net) and strategically integrates it at different architectural locations along with varied color features to specifically focus on intrinsic characteristics of small, littered, transparent and partially submerged bottles. The Swin transformer block uniquely processes features through a parallel path alongside the RepNCSPELAN block to concatenate multi-scale and attention features crucial for small waste bottle detection. The comparisons of the proposed work against 14 baseline YOLO variants and 16 state-of-the-art existing techniques indicate the proposed technique’s outperforming results with the highest precision, mean Average Precision (mAP@50) and mean of mAP (mmAP@50:95) of