A Transfer Learning Approach For Efficient Classification of Waste Materials

Md. Humaion Kabir Mehedi, Irfana Arafin, Murad Hasan, Farhin Rahman, Rufaida Tasin, Annajiat Alim Rasel

Abstract: The authors of this study have used the Waste Classification Dataset to build a highly accurate model that classified rubbish into two distinct groups in an effort to address the problem of waste classification for various classes of discarded material. VGG16, MobileNetV2, and a baseline 6 layer CNN model are used in the experiments. The VGG16 model have achieved 96.00% accuracy, while the MobileNetV2 model achieved 95.51 %, and the baseline CNN model achieved 90.61 % accuracy. The garbage in the input picture can be correctly classified by the neural network model. The experimental findings are compared to other studies in the same area. In addition, LIME is also implemented to make our models's prediction more explainable. This investigation's experimental applications are geared on facilitating more precise trash classification.