Mathematical Biosciences and Engineering (Jan 2023)
Research on deep learning garbage classification system based on fusion of image classification and object detection classification
Abstract
With the development of national economy, the output of waste is also increasing. People's living standards are constantly improving, and the problem of garbage pollution is increasingly serious, which has a great impact on the environment. Garbage classification and processing has become the focus of today. This topic studies the garbage classification system based on deep learning convolutional neural network, which integrates the garbage classification and recognition methods of image classification and object detection. First, the data sets and data labels used are made, and then the garbage classification data are trained and tested through ResNet and MobileNetV2 algorithms, Three algorithms of YOLOv5 family are used to train and test garbage object data. Finally, five research results of garbage classification are merged. Through consensus voting algorithm, the recognition rate of image classification is improved to 2%. Practice has proved that the recognition rate of garbage image classification has been increased to about 98%, and it has been transplanted to the raspberry pie microcomputer to achieve ideal results.
Keywords