Garbage Image Recognition and Classification Based on CNN

Yuhui Cao,Yan He,Jiaping Lin,Sihao Sun
DOI: https://doi.org/10.54254/2755-2721/4/20230507
2023-01-01
Abstract:The trend in today's society is to pursue a higher quality of life. Trash classificationn obviously needs to be taken into account. In reality, however, many people have problems with sorting, making them so inefficient that the time and effort spent on it outweighs the rewards. A model program based on a Convolutional Neural Network (CNN) machine learning approach is employed to perform the job of identifying and classifying to solve the problem. By dividing the data set into different uses and performing processes such as mirroring images, the recognition capability of the model is enhanced, and its accuracy can be verified. The number of layers is set at a reasonable level and various parameters are initialized. The learning rate set by the project adjusts itself to avoid overshooting and too low efficiency, but it also leads to fluctuations in cost during convergence. After a careful comparison of the pros and cons, the ReLu function was chosen as the activation function to ensure proper operation of the program and to speed up training and reduce computation. The experimental test results prove that the model outperforms its predecessors. The convolution of CNN compresses the data well, and the number of iterations ensures the accuracy of the model prediction. It could go even further, if more samples and more powerful hardware could be used to solve the problems of over-fitting and insufficient computing power.
What problem does this paper attempt to address?