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

被引:26
作者
Nguyen, Thanh-Hai [1 ]
Nguyen, Thanh-Nghia [1 ]
Ngo, Ba-Viet [1 ]
机构
[1] Ho Chi Minh City Univ Technol & Educ, Fac Elect Elect Engn, Ho Chi Minh City 700000, Vietnam
来源
AGRIENGINEERING | 2022年 / 4卷 / 04期
关键词
VGG-19; model; tomato leaf images; transfer learning; image segmentation; HSV color space; RECOGNITION; SUPERPIXEL;
D O I
10.3390/agriengineering4040056
中图分类号
S2 [农业工程];
学科分类号
0828 ;
摘要
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.
引用
收藏
页码:871 / 887
页数:17
相关论文
共 54 条
[1]   Development of Efficient CNN model for Tomato crop disease identification [J].
Agarwal, Mohit ;
Gupta, Suneet Kr ;
Biswas, K. K. .
SUSTAINABLE COMPUTING-INFORMATICS & SYSTEMS, 2020, 28 (28)
[2]   Less is More: Lighter and Faster Deep Neural Architecture for Tomato Leaf Disease Classification [J].
Ahmed, Sabbir ;
Hasan, Md Bakhtiar ;
Ahmed, Tasnim ;
Sony, Md Redwan Karim ;
Kabir, Md Hasanul .
IEEE ACCESS, 2022, 10 :68868-68884
[3]   AI-Driven Framework for Recognition of Guava Plant Diseases through Machine Learning from DSLR Camera Sensor Based High Resolution Imagery [J].
Almadhor, Ahmad ;
Rauf, Hafiz Tayyab ;
Lali, Muhammad Ikram Ullah ;
Damasevicius, Robertas ;
Alouffi, Bader ;
Alharbi, Abdullah .
SENSORS, 2021, 21 (11)
[4]   Automated disease classification in (Selected) agricultural crops using transfer learning [J].
Aravind, Krishnaswamy Rangarajan ;
Raja, Purushothaman .
AUTOMATIKA, 2020, 61 (02) :260-272
[5]   Digital image processing techniques for detecting, quantifying and classifying plant diseases [J].
Arnal Barbedo, Jayme Garcia .
SPRINGERPLUS, 2013, 2 :1-12
[6]   Deep Learning-Based Estimation of Crop Biophysical Parameters Using Multi-Source and Multi-Temporal Remote Sensing Observations [J].
Bahrami, Hazhir ;
Homayouni, Saeid ;
Safari, Abdolreza ;
Mirzaei, Sayeh ;
Mahdianpari, Masoud ;
Reisi-Gahrouei, Omid .
AGRONOMY-BASEL, 2021, 11 (07)
[7]   Deep Learning for Tomato Diseases: Classification and Symptoms Visualization [J].
Brahimi, Mohammed ;
Boukhalfa, Kamel ;
Moussaoui, Abdelouahab .
APPLIED ARTIFICIAL INTELLIGENCE, 2017, 31 (04) :299-315
[8]   Arabidopsis thaliana, an experimental host for tomato yellow leaf curl disease-associated begomoviruses by agroinoculation and whitefly transmission [J].
Canizares, M. C. ;
Rosas-Diaz, T. ;
Rodriguez-Negrete, E. ;
Hogenhout, S. A. ;
Bedford, I. D. ;
Bejarano, E. R. ;
Navas-Castillo, J. ;
Moriones, E. .
PLANT PATHOLOGY, 2015, 64 (02) :265-271
[9]   Multiple introductions of the old world Begomovirus Tomato yellow leaf curl virus into the new world [J].
Duffy, Siobain ;
Holmes, Edward C. .
APPLIED AND ENVIRONMENTAL MICROBIOLOGY, 2007, 73 (21) :7114-7117
[10]   A Robust Deep-Learning-Based Detector for Real-Time Tomato Plant Diseases and Pests Recognition [J].
Fuentes, Alvaro ;
Yoon, Sook ;
Kim, Sang Cheol ;
Park, Dong Sun .
SENSORS, 2017, 17 (09)