Intelligent system for solid waste classification using combination of image processing and machine learning models

Abstract

ABSTRACT Solid waste is a major issue in all cities around the world. Classification and segregation of solid waste prior to reuse, recycle or recover is an important step towards sustainable waste management. Traditional manual sorting of solid waste is a labour-intensive process that may pose health risks to the workers. Currently, automated classification of solid waste using machine learning techniques is widely applied. This study aims to develop an automated waste classification model by testing traditional and deep machine learning models. To achieve that, both open and generated datasets were used in the model training and testing. The study results showed relatively low prediction capability of the traditional machine learning models like Random Forest (RF) and Support Vector Machine (SVM) as compared to the deep machine learning Convolutional Neural Network (CNN). The testing of the three models on a combined data set of Trashnet with local garbage data set resulted in accuracy of 62.5% for SVM, 72.0% for RF and 92.7% for CNN. JONET deep learning model has been developed using a combination of pre-trained base model (DenseNet 201) with a new architicture that contains a fully connected layer in the classification stage with 1024 neurons. The model is capable to identify six classes of solid waste items with various accuracies. When tested on the Trashnet dataset, the accuracy was 96.06%, while testing on the local garbage dataset gave an accuracy of 94.40%. JONET has been tested also on multi object images which gave an acceptable prediction accuracy.