Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution

In image classification of tomato leaf diseases based on deep learning, models often focus on features such as edges, stems, backgrounds, and shadows of the experimental samples, while ignoring the features of the disease area, resulting in weak generalization ability. In this study, a self-attentio...

Full description

Bibliographic Details
Main Authors: Zheng Li, Weijie Tao, Jianlei Liu, Fenghua Zhu, Guangyue Du, Guanggang Ji
Format: Article
Language:English
Published: MDPI AG 2023-09-01
Series:Horticulturae
Subjects:
Online Access:https://www.mdpi.com/2311-7524/9/9/1034
_version_ 1797579818503503872
author Zheng Li
Weijie Tao
Jianlei Liu
Fenghua Zhu
Guangyue Du
Guanggang Ji
author_facet Zheng Li
Weijie Tao
Jianlei Liu
Fenghua Zhu
Guangyue Du
Guanggang Ji
author_sort Zheng Li
collection DOAJ
description In image classification of tomato leaf diseases based on deep learning, models often focus on features such as edges, stems, backgrounds, and shadows of the experimental samples, while ignoring the features of the disease area, resulting in weak generalization ability. In this study, a self-attention mechanism called GD-Attention is proposed, which considers global pixel value distribution information and guide the deep learning model to give more concern on the leaf disease area. Based on data augmentation, the proposed method inputs both the image and its pixel value distribution information to the model. The GD-Attention mechanism guides the model to extract features related to pixel value distribution information, thereby increasing attention towards the disease area. The model is trained and tested on the Plant Village (PV) dataset, and by analyzing the generated attention heatmaps, it is observed that the disease area obtains greater weight. The results achieve an accuracy of 99.97% and 27 MB parameters only. Compared to classical and state-of-the-art models, our model showcases competitive performance. As a next step, we are committed to further research and application, aiming to address real-world, complex scenarios.
first_indexed 2024-03-10T22:41:18Z
format Article
id doaj.art-7d630a7ee4bf40a395c11dc7262f96a7
institution Directory Open Access Journal
issn 2311-7524
language English
last_indexed 2024-03-10T22:41:18Z
publishDate 2023-09-01
publisher MDPI AG
record_format Article
series Horticulturae
spelling doaj.art-7d630a7ee4bf40a395c11dc7262f96a72023-11-19T10:59:17ZengMDPI AGHorticulturae2311-75242023-09-0199103410.3390/horticulturae9091034Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value DistributionZheng Li0Weijie Tao1Jianlei Liu2Fenghua Zhu3Guangyue Du4Guanggang Ji5School of Rail Transportation, Shandong Jiaotong University, Jinan 250357, ChinaSchool of Rail Transportation, Shandong Jiaotong University, Jinan 250357, ChinaDepartment of Cyberspace Security, Qufu Normal University, Qufu 273165, ChinaInstitute of Automation, Chinese Academy of Sciences, Beijing 100049, ChinaSchool of Rail Transportation, Shandong Jiaotong University, Jinan 250357, ChinaSchool of Rail Transportation, Shandong Jiaotong University, Jinan 250357, ChinaIn image classification of tomato leaf diseases based on deep learning, models often focus on features such as edges, stems, backgrounds, and shadows of the experimental samples, while ignoring the features of the disease area, resulting in weak generalization ability. In this study, a self-attention mechanism called GD-Attention is proposed, which considers global pixel value distribution information and guide the deep learning model to give more concern on the leaf disease area. Based on data augmentation, the proposed method inputs both the image and its pixel value distribution information to the model. The GD-Attention mechanism guides the model to extract features related to pixel value distribution information, thereby increasing attention towards the disease area. The model is trained and tested on the Plant Village (PV) dataset, and by analyzing the generated attention heatmaps, it is observed that the disease area obtains greater weight. The results achieve an accuracy of 99.97% and 27 MB parameters only. Compared to classical and state-of-the-art models, our model showcases competitive performance. As a next step, we are committed to further research and application, aiming to address real-world, complex scenarios.https://www.mdpi.com/2311-7524/9/9/1034plant leaf diseaseimage recognitionattention mechanismsmart agriculture
spellingShingle Zheng Li
Weijie Tao
Jianlei Liu
Fenghua Zhu
Guangyue Du
Guanggang Ji
Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
Horticulturae
plant leaf disease
image recognition
attention mechanism
smart agriculture
title Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
title_full Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
title_fullStr Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
title_full_unstemmed Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
title_short Tomato Leaf Disease Recognition via Optimizing Deep Learning Methods Considering Global Pixel Value Distribution
title_sort tomato leaf disease recognition via optimizing deep learning methods considering global pixel value distribution
topic plant leaf disease
image recognition
attention mechanism
smart agriculture
url https://www.mdpi.com/2311-7524/9/9/1034
work_keys_str_mv AT zhengli tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution
AT weijietao tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution
AT jianleiliu tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution
AT fenghuazhu tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution
AT guangyuedu tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution
AT guanggangji tomatoleafdiseaserecognitionviaoptimizingdeeplearningmethodsconsideringglobalpixelvaluedistribution