Waste Classification using Transfer Learning with Convolutional Neural Networks

K Srivatsan,Surender Dhiman,Anuj Jain
DOI: https://doi.org/10.1088/1755-1315/775/1/012010
2021-06-01
IOP Conference Series: Earth and Environmental Science
Abstract:Abstract With the aim to tackle the issue of waste classification for different categories of misspend substances, the authors, with a limited availability of dataset have processed a highly accurate model to classify garbage into 7 different categories using the CompostNet dataset. Experiments were carried out on pre-trained models of MobileNetV2, ResNet34 and Densenet121 model, previously trained on ImageNet dataset. The accuracies obtained were 96.42%, 96.27% and 96.273% respectively for the Densenet121, mobilenetv2 and resnet34 models. Within 60 epochs, the neural network model accurately categorizes waste materials provided in the input image. The results of the experiments are compared with other previous work done in the same field. The applications of the experiments conducted in this research aims at providing better waste categorization and also follows the United Nations goal for Responsible Consumption and Production towards sustainable development.
What problem does this paper attempt to address?