An Irregularly Dropped Garbage Detection Method Based on Improved YOLOv5s

Abstract

Waste sorting and recycling play a significant role in carbon neutrality, and the government has promoted waste sorting stations in various cities while the stations have limited efficiency due to the absence of intelligent surveillance systems to monitor and analyze the scene in waste stations, especially to detect the irregularly dropped garbage. To take the most advantage of these stations, this study proposes an improved YOLO (You Only Look Once) v5s detector named YOLOv5s-Garbage to monitor waste sorting stations in real-time. This study enhances its ability to detect garbage by introducing CBAM (Convolutional Block Attention Module) and using EIoU (Efficient Intersection over Union) to accelerate the convergence of the bonding box loss. According to experiments, the mAP of YOLOv5s-Garbage on the waste sorting dataset reaches 89.7%, which is 3.3% higher than the classical YOLOv5s. This study then combines the DeepSort tracking algorithm and re-filter process to filter the target garbage to distinguish the irregularly dropped garbage and normal one, which reduces the false alarm significantly.