Skip to main content

Research Repository

Advanced Search

GANS-based data augmentation for citrus disease severity detection using deep learning

Zeng, Qingmao; Ma, Xinhui; Cheng, Baoping; Zhou, Erxun; Pang, Wei


Qingmao Zeng

Baoping Cheng

Erxun Zhou

Wei Pang


Recently, many Deep Learning models have been employed to classify different kinds of plant diseases, but very little work has been done for disease severity detection. However, it is more important to master the severities of plant diseases accurately and timely, as it helps to make effective decisions to protect the plants from being further infected and reduce financial loss. In this paper, based on the Huanglongbing (HLB)-infected leaf images obtained from PlantVillage and crowdAI, we created a dataset with 5,406 citrus leaf images infected by HLB. Then six different kinds of popular models were trained to perform the severity detection of citrus HLB with the goal to find which types of models are more suitable to detect HLB severity with the same training circumstance. The experimental results show that the Inception_v3 model with epochs=60 can achieve higher accuracy than that of other models for severity detection with an accuracy of 74.38% due to its highly computational efficiency and small number of parameters. Additionally, aiming for evaluating whether GANs-based data augmentation can contribute to improve the model learning performance, we adopted DCGANs (Deep Convolutional Generative Adversarial Networks) to augment the original training dataset up to two times itself. Finally, a new training dataset with 14,056 leaf images composed by the original training images and the augmented ones were used to train the Inception_v3 model. As a result, we achieved an accuracy of 92.60%, about 20% higher than that of the Inception_v3 model trained by the original training dataset, which suggested that the GANs-based data augmentation is very useful to improve the model learning performance.


Zeng, Q., Ma, X., Cheng, B., Zhou, E., & Pang, W. (2020). GANS-based data augmentation for citrus disease severity detection using deep learning. IEEE Access, 8, 172882-172891.

Journal Article Type Article
Acceptance Date Sep 8, 2020
Online Publication Date Sep 18, 2020
Publication Date 2020
Deposit Date Nov 2, 2020
Publicly Available Date Nov 2, 2020
Journal IEEE Access
Electronic ISSN 2169-3536
Publisher Institute of Electrical and Electronics Engineers
Peer Reviewed Peer Reviewed
Volume 8
Pages 172882-172891
Keywords Citrus Huanglongbing; Data augmentation; Deep learning; Generative adversarial networks; Plant disease severity
Public URL
Publisher URL


You might also like

Downloadable Citations