Click on “Download PDF” for the PDF version or on the title for the HTML version.


If you are not an ASABE member or if your employer has not arranged for access to the full-text, Click here for options.

A Data Augmentation Approach Based on Generative Adversarial Networks for Date Fruit Classification

Published by the American Society of Agricultural and Biological Engineers, St. Joseph, Michigan www.asabe.org

Citation:  Applied Engineering in Agriculture. 38(6): 975-982. (doi: 10.13031/aea.15107) @2022
Authors:   Donald Ufuah, Gabriel Thomas, Simone Balocco, Annamalai Manickavasagan
Keywords:   Data augmentation, Deep learning, Expert system, Fruit classification, GAN, Neural networks.

Highlights

Classification of date fruit hardness levels from images.

Combination of deep networks and an expert system to yield better accuracies.

Abstract. Machine learning techniques have been used in various agricultural applications from farming to post-harvest operations. For some cases, a large amount of data is not available and improving on classification accuracies based on deep networks cannot be an option. Such is the case presented here for sorting date fruits based on their hardness into three classes (soft, semi-hard, hard). The original dataset in this work consists of 1800 monochrome images with 600 images per class obtained from different growing regions in Oman. This is a limited number of examples to consider deep networks. Thus, this work proposes data augmentation based on Generative Adversarial Networks (GAN) to synthetically augment the date fruit images. It incorporates a Densely Connected Convolutional Network (DenseNet) for date fruit classification. The goal is to generate enough images so that DenseNet can successfully classify them. The GAN images help during the training part when a large dataset is used for training. Accuracies of 100%, 100%, and 99% were achieved for the hard, soft, and semi-hard classes by using an expert system in combination with two DenseNet networks, one network trained for classifying hard and soft cases as well as a second network that classifies the three classes.

(Download PDF)    (Export to EndNotes)