A VGG-19 Model with Transfer Learning and Image Segmentation for Classification of Tomato Leaf Disease

• Published in: AgriEngineering Vol. 4; no. 4; pp. 871 - 887
• Main Authors: Nguyen, Thanh-Hai, Nguyen, Thanh-Nghia, Ngo, Ba-Viet
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.12.2022
• Subjects: Accuracy; Algorithms; Artificial intelligence; Artificial neural networks; Classification; Crop diseases; Datasets; Deep learning; HSV color space; Image classification; Image processing; Image segmentation; Internet of Things; Learning; Leaves; Machine learning; Medical imaging; Model accuracy; Neural networks; Plant diseases; Saturation (color); tomato leaf images; Tomatoes; Training; Transfer learning; VGG-19 model
• ISSN: 2624-7402; 2624-7402
• DOI: 10.3390/agriengineering4040056

Tomato leaves can have different diseases which can affect harvest performance. Therefore, accurate classification for the early detection of disease for treatment is very important. This article proposes one classification model, in which 16,010 tomato leaf images obtained from the Plant Village database are segmented before being used to train a deep convolutional neural network (DCNN). This means that this classification model will reduce training time compared with that of the model without segmenting the images. In particular, we applied a VGG-19 model with transfer learning for re-training in later layers. In addition, the parameters such as epoch and learning rate were chosen to be suitable for increasing classification performance. One highlight point is that the leaf images were segmented for extracting the original regions and removing the backgrounds to be black using a hue, saturation, and value (HSV) color space. The segmentation of the leaf images is to synchronize the black background of all leaf images. It is obvious that this segmentation saves time for training the DCNN and also increases the classification performance. This approach improves the model accuracy to 99.72% and decreases the training time of the 16,010 tomato leaf images. The results illustrate that the model is effective and can be developed for more complex image datasets.