Research on Garbage Classification Algorithm Based on Machine Learning

With the development of urbanization and the improvement of consumption levels, garbage disposal has become one of the major challenges facing environmental protection. In order to classify garbage efficiently and accurately and promote the construction of “two-oriented society” and smart cities, this topic studies a garbage classification technology based on machine learning. Based on the PyTorch framework design, I compared the training and testing of the AlexNet model and the VGG model. Through testing the experimental results, it was found that using the VGG19 model and optimization method, the accuracy of garbage classification can reach 95%, which is higher than other models. Finally, the VGG19 model is used for training and transfer learning.