Application of Transfer Learning for Fruits and Vegetable Quality Assessment

Sherzod Turaev, Ali Abd Almisreb, Mohammed A. Saleh

Research output: Chapter in Book/Report/Conference proceedingConference contribution

6 Citations (Scopus)

Abstract

In this paper, we utilize the concept of transfer learning in fruits and vegetable quality assessment. The transfer learning concept applies the idea of reuse the pre-trained Convolutional Neural Network to solve a new problem without the need for large-scale datasets for training. Eight pre-trained deep learning models namely AlexNet, GoogleNet, ResNet18, ResNet50, ResNet101, Vgg16, Vgg19, and NasNetMobile are fine-tuned accordingly to evaluate the quality of fruits and vegetable. To evaluate the training and validation performance of each fine-tuned model, we collect a dataset consists of images from 12 fruits and vegetable samples. The dataset builds over five weeks. For every week 70 images collected therefore the total number of images over five weeks is 350 and the total number of images in the dataset is (12∗350) 4200 images. The overall number of classes in the dataset is (12∗5) 60 classes. The evaluation of the models was conducted based on this dataset and also based on an augmented version. The model's outcome shows that the Vgg19 model achieved the highest validation accuracy over the original dataset with 91.50% accuracy and the ResNet18 model scored the highest validation accuracy based on the augmented dataset with 91.37% accuracy.

Original languageEnglish
Title of host publicationProceedings of the 2020 14th International Conference on Innovations in Information Technology, IIT 2020
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages7-12
Number of pages6
ISBN (Electronic)9781728181844
DOIs
Publication statusPublished - Nov 17 2020
Event14th International Conference on Innovations in Information Technology, IIT 2020 - Al Ain, United Arab Emirates
Duration: Nov 17 2020Nov 18 2020

Publication series

NameProceedings of the 2020 14th International Conference on Innovations in Information Technology, IIT 2020

Conference

Conference14th International Conference on Innovations in Information Technology, IIT 2020
Country/TerritoryUnited Arab Emirates
CityAl Ain
Period11/17/2011/18/20

Keywords

  • Transfer learning
  • assessment
  • deep learning
  • fruits
  • vegetable

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Science Applications
  • Information Systems
  • Information Systems and Management
  • Safety, Risk, Reliability and Quality

Fingerprint

Dive into the research topics of 'Application of Transfer Learning for Fruits and Vegetable Quality Assessment'. Together they form a unique fingerprint.

Cite this