Smart GreenGrocer: Automatic Vegetable Type Classification Using the CNN Algorithm

Abstract

In the food industry, separating vegetables is done by visually trained professionals. However, because it takes plenty of time to sort a large number of different types of vegetables, human errors might arise at any time, and using human resources is not always effective. Thus, automation is needed to minimize process time and errors. Computer vision helps reduce the need for human resources by automatizing the classification. Vegetables come in various colors and shapes; thus, vegetable classification becomes a challenging multiclass classification due to intraspecies variety and interspecies similarity of these main distinguishing characteristics. Consequently, much research is made to automatically discover effective methods to group each type of vegetable using computers. To answer this challenge, we proposed a solution utilizing deep learning with a Convolutional Neural Network (CNN) to perform multi-label classification on some types of vegetables. We experimented with the modification of batch size and optimizer type. In the training process, the learning rate is 0.01, and it adapts on arrival in the local minimum for result optimization. This classification is performed on 15 types of vegetables and produces 98.1% accuracy on testing data with 25 minutes and 45 seconds of training time.