Comparison of Machine Learning Approach for Waste Bottle Classification

Abstract

The use of machine learning for the image classification process is growing all the time. Many methods can be used to classify an image with good accuracy. Convolutional Neural Network (CNN) and Support Vector Machine (SVM) are popular methods for this case. The two approaches have differences in the data training process to achieve classification objectives. Although there are some differences between these approaches, there are some advantages to both of them. This research explores the comparison of the two CNN and SVM methods by comparing the training process carried out and the accuracy results of the classification. The process stages are divided into pre-processing, training, and testing. The objects used are ten waste plastic bottles with different brands of medium size with a total data of 1100 images. Based on the observations, both methods have advantages and disadvantages in the data training and classification process. However, from the results, CNN's accuracy is better than SVM. The accuracy of both networks is 99% for CNN and 74% for SVM, respectively. So, from the results of experiments that have been carried out in the study, it was found that CNN was still better than SVM. Doi: 10.28991/ESJ-2022-06-05-011 Full Text: PDF