DCNN-BasedVegetableImageClassificationUsingTransferLearning AComparativeStudy
DCNN-BasedVegetableImageClassificationUsingTransferLearning AComparativeStudy
net/publication/352846889
CITATIONS READS
5 3,775
3 authors:
SEE PROFILE
All content following this page was uploaded by M. Israk Ahmed on 17 July 2021.
Abstract—In this paper, an attempt is addressed towards accu- As several steps of vegetable production to consumption is
rate vegetable image classification. A dataset consisting of 21,000 still depended on manual operation with a huge sum of labor
images of 15 classes is used for this classification. Convolutional constrain, it truly influences the advancement of commercial-
neural network, a deep learning algorithm is the most efficient
tool in the machine learning field for classification problems. But ization of vegetable items. To solve this issue, automation in
CNN requires large datasets so that it performs well in natural vegetable picking, sorting, labeling is required by introducing
image classification problems. Here, we conduct an experiment a vegetable image classifier so that time and money can be
on the performance of CNN for vegetable image classification saved. In modern days, in the field of agriculture, fundamental
by developing a CNN model from the ground. Additionally, research work are classification and detection. As there are
several pre-trained CNN architectures using transfer learning
are employed to compare the accuracy with the typical CNN. various sorts of vegetables and numerous individuals don’t
This work proposes the study between such typical CNN and its have any idea about them. So, the plan of a vegetable classifier
architectures(VGG16, MobileNet, InceptionV3, ResNet etc.) to will likewise carry straightforwardness to individuals lives.
build up which technique would work best regarding accuracy Also, the sorting of vegetables is manually done in super shops
and effectiveness with new image datasets. Experimental results and distribution centers. Therefore, to solve these problems
are presented for all the proposed architectures of CNN. Besides,
a comparative study is done between developed CNN models this research is conducted. The purpose of this paper is
and pre-trained CNN architectures. And the study shows that to classify vegetable images with higher accuracy, with the
by utilizing previous information gained from related large- help of CNN and pre-trained DCNN with transfer learning.
scale work, the transfer learning technique can achieve better Nowadays, convolutional neural network is often used for clas-
classification results over traditional CNN with a small dataset. sification, segmentation, image recognition etc. Deep network
And one more enrichment in this paper is that we build up a
vegetable images dataset of 15 categories consisting of a total of architecture is the utmost power of CNN which enables CNN
21,000 images. to automatically learn mid to high-level considerations from
Index Terms—Vegetable image classification, deep learning, new data [2], [3]. In this research, we proposed a CNN model
CNN, VGG16, MobileNet, Inception-V3, ResNet. developed from scratch as well as four fine-tuned state-of-
the-art CNN architectures (InceptionV3, VGG16, Resnet, Mo-
I. I NTRODUCTION bileNet) for vegetable image classification. Also a comparative
Vegetables are one of the most common food items in study is done based on the performance between CNN and its
everyday meals worldwide. People around the world produce architectures.
many kinds of vegetables. On our planet there exists almost
hundreds of thousands of species of vegetables according to II. R ELATED W ORK
some survey [1]. And vegetables are important for human This research is about classifying vegetable images with
beings as a result of their nutrients, minerals, phytochemical higher accuracy and efficiency. And also finding out the effi-
mixtures, and dietary fiber content. There are similarities in cient model along with a technique that performs well in terms
many vegetable types in terms of color, texture, and shape. of accuracy, time, and cost. Almost 3 years ago, Om Patil et
Even country-wise same vegetable has a different name. From al. [4] used InceptionV3( known as GoogLeNet ) for vegetable
vegetable production to delivery, several common steps are classification tasks. By fine-tuning the inception network and
operated manually. Like picking, and sorting vegetables. And applying the transfer learning technique, the proposed model
recognizing a vegetable is a hard task for the customer in the can classify 4 types of vegetables- carrots, onions, cucumbers,
market, as there are similarities between different vegetables. and tomatoes. The accuracy of their fine-tuned inception-V3 is
TABLE II
E XISTING METHODS , DATASETS , AND RESULTS