Deep Learning-Based Plant-Image Classification Using a Small Training Dataset

Research output: Contribution to journalArticlepeer-review

21 Scopus citations

Abstract

Extensive research has been conducted on image augmentation, segmentation, detection, and classification based on plant images. Specifically, previous studies on plant image classification have used various plant datasets (fruits, vegetables, flowers, trees, etc., and their leaves). However, existing plant-based image datasets are generally small. Furthermore, there are limitations in the construction of large-scale datasets. Consequently, previous research on plant classification using small training datasets encountered difficulties in achieving high accuracy. However, research on plant image classification based on small training datasets is insufficient. Accordingly, this study performed classification by reducing the number of training images of plant-image datasets by 70%, 50%, 30%, and 10%, respectively. Then, the number of images was increased back through augmentation methods for training. This ultimately improved the plant-image classification performance. Based on the respective preliminary experimental results, this study proposed a plant-image classification convolutional neural network (PI-CNN) based on plant image augmentation using a plant-image generative adversarial network (PI-GAN). Our proposed method showed the higher classification accuracies compared to the state-of-the-art methods when the experiments were conducted using four open datasets of PlantVillage, PlantDoc, Fruits-360, and Plants.

Original languageEnglish
Article number3091
JournalMathematics
Volume10
Issue number17
DOIs
StatePublished - Sep 2022

Keywords

  • PI-CNN
  • PI-GAN
  • deep learning
  • image augmentation
  • plant image classification

Fingerprint

Dive into the research topics of 'Deep Learning-Based Plant-Image Classification Using a Small Training Dataset'. Together they form a unique fingerprint.

Cite this