Machine learning and IoT-based garbage detection system for smart cities

Abstract

Today, detecting waste, collecting it, processing it, and getting rid of it are among the most significant environmental issues in developing and undeveloped counties. It has been observed that a large amount of garbage remains strewn on the roadside. This study presented a garbage detection technology such as machine learning and gadgets connected to the Internet of Things (IoT), such as an IP-enabled CCTV camera, to take pictures and send them to the city’s main server. The input images are transformed into a two-dimension array of integers using Python modules and divided into the garbage and no garbage classes. There is an 80:20 split between the training and testing datasets from the input dataset. Preprocessed images are then utilised as inputs for a wide range of machine learning and neural network models for classification; these include K-Nearest Neighbour (KNN), Logistic Regression (LR), Naive Bayes (NB), and Support Vector Machine (SVM). The test data sets are applied, and a confusion matrix is formed for all models to analyse the efficiency and performance of the trained models. Results from the confusion matrix are contrasted with those from the area under the Receiver characteristics operating curve (AUC). As a result, the ConvNet model is best suited for classifying garbage or no garbage present in open space, and the LR model proposed best suits the garbage detection problem. The proposed models are best suitable for improving the efficiency of existing garbage identification systems and developing a new system for smart cities.