Grape leaf disease identification with sparse data via generative adversarial networks and convolutional neural networks

Yiping Chen,Qiufeng Wu
DOI: https://doi.org/10.1007/s11119-022-09941-z
IF: 5.767
2022-08-09
Precision Agriculture
Abstract:The main challenge in deep learning related to the identification of grape leaf diseases is how to achieve good performance in the case of available sparse datasets or limited number of annotated samples, small lesions, redundant information and blurred background information in grape leaf disease images. This paper proposes a three-stage deep learning-based pipeline, including a convolutional neural netword (Faster R-CNN) for detection of lesions, a generative adversarial network (DCGAN) for data augmentation and a residual neural network (ResNet) for identification of lesions, to solve these problems. Firstly, Faster R-CNN was used to mark the location of grape leaf lesions to obtain lesions dataset for data augmentation and ResNet for identification of lesions. Secondly, leaf lesion images were fed into DCGAN to generate synthetic grape lesion images for identification of lesions. Finally, ResNet trained in the training dataset consisting of real grape leaf lesions and synthetic grape lesion images obtained by DCGAN, was used to identify the grape leaf lesions according to the majority voting principle. The first experimental results showed that Faster R-CNN, DCGAN and ResNet had good performance in each stage. Secondly, the second experimental results showed that the proposed three-stage method is superior to single-stage and two-stage methods in the case of sparse datasets and small lesions. Finally, a generalization experiment was also carried out using 100 images on the internet to verify the generalization of the proposed three-stage method. Experimental results showed that the proposed three-stage method had good generalization ability.
agriculture, multidisciplinary
What problem does this paper attempt to address?