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

• Published in: Sensors (Basel, Switzerland) Vol. 24; no. 14; p. 4666
• Main Authors: Liu, Shenlin, Chen, Ruihan, Ye, Minhua, Luo, Jiawei, Yang, Derong, Dai, Ming
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 18.07.2024
• Subjects: BiFPN; Cameras; CBAM; Datasets; domestic waste; Initiatives; intricate environmental landscapes; lightweight; Neural networks; P2; domestic waste; CBAM; BiFPN; intricate environmental landscapes; lightweight
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s24144666

In response to the challenges of accurate identification and localization of garbage in intricate urban street environments, this paper proposes EcoDetect-YOLO, a garbage exposure detection algorithm based on the YOLOv5s framework, utilizing an intricate environment waste exposure detection dataset constructed in this study. Initially, a convolutional block attention module (CBAM) is integrated between the second level of the feature pyramid etwork (P2) and the third level of the feature pyramid network (P3) layers to optimize the extraction of relevant garbage features while mitigating background noise. Subsequently, a P2 small-target detection head enhances the model's efficacy in identifying small garbage targets. Lastly, a bidirectional feature pyramid network (BiFPN) is introduced to strengthen the model's capability for deep feature fusion. Experimental results demonstrate EcoDetect-YOLO's adaptability to urban environments and its superior small-target detection capabilities, effectively recognizing nine types of garbage, such as paper and plastic trash. Compared to the baseline YOLOv5s model, EcoDetect-YOLO achieved a 4.7% increase in mAP , reaching 58.1%, with a compact model size of 15.7 MB and an FPS of 39.36. Notably, even in the presence of strong noise, the model maintained a mAP exceeding 50%, underscoring its robustness. In summary, EcoDetect-YOLO, as proposed in this paper, boasts high precision, efficiency, and compactness, rendering it suitable for deployment on mobile devices for real-time detection and management of urban garbage exposure, thereby advancing urban automation governance and digital economic development.