Deep Neural Networks for Garbage Classification
Effective garbage classification and recycling are of great significance for protecting the environment, conserving resources and improving people's livelihoods, and have a far-reaching impact on promoting sustainable development. However, the traditional waste classification method has problems such as low accuracy and low efficiency, while its reliance on manual labor and high time consumption make it difficult to adapt to the needs of modern society for waste classification. To address this problem, we propose in this study to utilize deep convolutional neural networks, mainly Residual Network with 18 layers (ResNet18) model to improve the efficiency and accuracy of garbage classification. In this study, ResNet18 is first evaluated comprehensively for several key performance metrics in the waste classification task, including: Accuracy (Accuracy, Recall, Precision, F1-score, Kappa coefficient, Params and FLOPs. Then, we also evaluate the performance of classical convolutional neural network models such as LeNet, Visual Geometry Group Network with 19 layers (VGG19), AlexNet, and MobileNet, and analyze them in comparison with ResNet18, which shows that ResNet18 has a superior performance in dealing with the waste classification The results show that ResNet18 has superior performance in the task of garbage classification. Finally, we analyze the reasons why ResNet18 exhibits the best performance and propose directions for further optimizing the model performance, which provide directions for future research and experiments.
Authors
- Hong‐Li Ren (ORCID: https://orcid.org/0000-0001-7194-0567)
- Haodong Liu (ORCID: https://orcid.org/0000-0001-8533-0882)
- Jiayao Qin
Institutions
- Taiyuan University of Technology (CN)
Publication Details
- Published
- 2024-09-20
- DOI
- https://doi.org/10.1109/iiotbdsc64371.2024.00013
- Citations
- 1
- Primary Topic
- Municipal Solid Waste Management
- Type
- article
- Field-Weighted Citation Impact
- 0.29