A Garbage Classification Method Based on Improved YOLOv5
Xiaobo Yan, Yuan Yang, Feng Le, Lin Wang, Mian Tan · 2022
With the development of national economy, people's daily garbage is increasing day by day. Relying on manpower to sort garbage is a heavy workload and low efficiency. In this paper, an automatic garbage classification system model based on computer vision is proposed to solve the above problems. Firstly, the YOLOv5 object detection algorithm is improved, and abandoning small object detection to obtain faster detection speed. Secondly, shortcut the complex network layer in the YOLOv5 algorithm framework to speed up the recognition speed while ensuring the recognition accuracy. Finally, the traditional loss function is used to improve the detection speed of large objects in raspberry pie. We collect various types of garbage data set, trains the model on the improved algorithm and tests it by embedding raspberry pie. The experimental results show that the proposed algorithm framework has the characteristics of fast speed and high precision, which can complete the garbage classification at the source of garbage.