Classification of tobacco leaf pests using VGG16 transfer learning

Dwiretno Istiyadi Swasono, Handayani Tjandrasa, Chastine Fathicah

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

48 Citations (Scopus)

Abstract

Some of the tobacco leaf pest attacks were only seen after the initial fermentation process. Tobacco leaves affected by pest attacks make the quality decline. Leaves affected by pests and diseases need to be separated from healthy leaves to maintain quality. Sorting is usually done manually allowing errors due to human-errors. In this study, we tried to classify the leaves affected by several types of pest attacks automatically. Convolutional Neural Network (CNN) is one of the latest classification methods proposed in this study using the famous VGG16 architecture. VGG16 training can last a long time if trained with random initialization of weights. For this reason, we selected initial weights by transfer learning to improve accuracy and speed up training time. Based on the results of training with 3-classes of the diseases using VGG16 and transfer learning, we obtained a very high accuracy. Some scenarios are tested based on a combination of the number of learnable parameters and types of the optimizer to get the best results. The result was that the proposed architecture was proven to be able to classify all training and validation data correctly. The dataset used was 1500 total images with 20% random cross-validation.

Original languageEnglish
Title of host publicationProceedings of 2019 International Conference on Information and Communication Technology and Systems, ICTS 2019
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages176-181
Number of pages6
ISBN (Electronic)9781728121338
DOIs
Publication statusPublished - Jul 2019
Event12th International Conference on Information and Communication Technology and Systems, ICTS 2019 - Surabaya, Indonesia
Duration: 18 Jul 2019 → …

Publication series

NameProceedings of 2019 International Conference on Information and Communication Technology and Systems, ICTS 2019

Conference

Conference12th International Conference on Information and Communication Technology and Systems, ICTS 2019
Country/TerritoryIndonesia
CitySurabaya
Period18/07/19 → …

Keywords

  • Tobacco Leaf Pest
  • Transfer Learning
  • VGG16

Fingerprint

Dive into the research topics of 'Classification of tobacco leaf pests using VGG16 transfer learning'. Together they form a unique fingerprint.

Cite this