A Garbage Classification Method Based on a Small Convolution Neural Network

Abstract

To improve the efficiency of social garbage classification, a garbage classification method based on a small convolutional neural network (CNN) is proposed in this paper. For low accuracy caused by light and shadow interference, an adaptive image-brightening algorithm is developed to average the brightness of the background in the image preprocessing stage, and a threshold replacement method is used to reduce shadow noise. Then, the Canny operator is used to assist in cropping the blank background in the image. For debugging low efficiency caused by the complex network, the neural network is optimized based on the MLH-CNN model to make its results simpler and equally efficient. Experimental results show the preprocessing in this study can improve the accuracy of model garbage classification. The CNN model in this study can achieve an accuracy of 96.77% on the self-built dataset and 93.72% on the TrashNet dataset, which is higher than the 92.6% accuracy of the MLC-CNN model. The network optimizer can also enhance the classification ability of the network model using the Adamax optimization algorithm based on Adam variants. In this paper, the network model derived from training is combined with the host computer software to design a garbage detection page so the model has a wider range of uses, which has a good effect on promoting the development of social environmental protection and improving residents’ awareness of environmental protection.