IJCATR Volume 9 Issue 1

Research on Garbage Image Classification Method Based on Convolutional Neural Network

Kangjian Tang, Zhan Wen, Yahui Chen, Wenzao Li
10.7753/IJCATR0901.1007
keywords : Deep learning; Convolutional neural network; Inception

PDF
The issue of garbage classification has aroused great concern for residents recently. The traditional garbage classification method has a low classification accuracy rate, high cost, and large personnel demand. With the increasing amount of garbage, it is difficult for traditional methods to effectively separate waste. Therefore, consider applying deep learning to the garbage classification problem. This paper uses two types of Convolutional Neural Networks (CNNs)--Inception V3 and Inception V4 to train Huawei's public garbage data set (Garbage Date) and establish a garbage classification model. After experiments, the classification results are compared and the performance of the model is tested. In this paper, by observing the changes in the accuracy rate and cross-entropy loss function of the two models on the training and test sets in the experiment, it is found that both models can obtain higher accuracy of garbage classification The network model using Inception V4 is more stable and accurate than the network model using Inception V3. The experimental results also show that this method can improve the accuracy of garbage classification in daily life and the efficiency of recyclable garbage collection.
@artical{k912020ijcatr09011007,
Title = "Research on Garbage Image Classification Method Based on Convolutional Neural Network",
Journal ="International Journal of Computer Applications Technology and Research(IJCATR)",
Volume = "9",
Issue ="1",
Pages ="37 - 40",
Year = "2020",
Authors ="Kangjian Tang, Zhan Wen, Yahui Chen, Wenzao Li "}
  • This paper proposes a new method for garbage image classification
  • Establish a garbage image classification model
  • This article compares the advantages and disadvantages of two convolutional neural network
  • Use a single picture set to test network performance.