Liu Shenlin, Chen Ruihan, Ye Minhua, Luo Jiawei, Yang Derong, Dai Ming
School of Mathematics and Computer, Guangdong Ocean University, Zhanjiang 524008, China.
Artificial Intelligence Research Institute, International (Macau) Institute of Academic Research, Macau 999078, China.
Sensors (Basel). 2024 Jul 18;24(14):4666. 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.
针对复杂城市街道环境中垃圾精确识别与定位的挑战,本文提出了EcoDetect-YOLO,一种基于YOLOv5s框架的垃圾暴露检测算法,利用本研究构建的复杂环境垃圾暴露检测数据集。首先,在特征金字塔网络(P2)的第二层和特征金字塔网络(P3)的第三层之间集成了卷积块注意力模块(CBAM),以优化相关垃圾特征的提取,同时减轻背景噪声。随后,一个P2小目标检测头提高了模型识别小垃圾目标的效率。最后,引入双向特征金字塔网络(BiFPN)来增强模型的深度特征融合能力。实验结果表明,EcoDetect-YOLO对城市环境具有适应性,并且具有卓越的小目标检测能力,能够有效识别纸张和塑料垃圾等九种类型的垃圾。与基线YOLOv5s模型相比,EcoDetect-YOLO的平均精度均值(mAP)提高了4.7%,达到58.1%,模型紧凑,大小为15.7MB,帧率为39.36。值得注意的是,即使在存在强噪声的情况下,该模型的mAP仍超过50%,凸显了其鲁棒性。综上所述,本文提出的EcoDetect-YOLO具有高精度、高效率和紧凑性,适合部署在移动设备上用于城市垃圾暴露的实时检测与管理,从而推动城市自动化治理和数字经济发展。