Intelligent classification trash bin based on deep learning

摘要

In order to solve the problems of low accuracy and low labor efficiency in garbage classification, an intelligent classification trash bin equipped with robot arm and 110-degree wide-angle camera is designed. A fast identification and positioning algorithm based on deep learning is proposed. The PASCAL VOC 2007 dataset is applied, which consists of 600 images containing 4000 different types of garbage in different situations. 20% of the images is randomly selected as the validation set, and the remaining 80% as the training set. The dataset is trained using the YOLOv5 model. After training, the model is tested on the garbage can system, automatically identification and classification the type of garbage is achieved and the robotic arm realizes the automatic positioning, grabbing and sorting. The experimental test results show that the error of the positioning algorithm is less than 0.8 mm, the fastest recognition speed is 2.6ms/piece, and the recognition accuracy can be maintained above 90% in the experimental environment, indicating that the algorithm has good accuracy and stability.