Garbage Recognition and Classification System Based on Convolutional Neural Network VGG16

Wang Hao,Hao Wang
DOI: https://doi.org/10.1109/aemcse50948.2020.00061
2020-04-01
Abstract:To study the application of deep learning in the field of environmental protection, the convolutional neural network VGG16 model is used to solve the problem of identification and classification of domestic garbage. This solution first used the OpenCV computer vision library to locate and select the identified objects and preprocessed the images into $224\times 224$ pixel RGB images accepted by the VGG16 network. Then after data enhancement, a VGG16 convolutional neural network based on the TensorFlow framework is built, by using the RELU activation function and adding BN layer to accelerate the model's convergence speed, while ensuring recognition accuracy. This project finally classifies domestic garbage into recyclable garbage, hazardous garbage, kitchen waste and other garbage. After actual tests, the correct classification rate of the garbage classification system based on VGG16 network proposed in this paper is 75.6%, the result meets the needs of daily use.
What problem does this paper attempt to address?