Sample Expansion and Classification Model of Maize Leaf Diseases Based on the Self-Attention CycleGAN

Hongliang Guo, Mingyang Li, Ruizheng Hou, Hanbo Liu, Xudan Zhou, Chunli Zhao, Xiao Chen () and Lianxing Gao ()
Additional contact information
Hongliang Guo: College of Information Technology, Jilin Agricultural University, Changchun 130118, China
Mingyang Li: College of Information Technology, Jilin Agricultural University, Changchun 130118, China
Ruizheng Hou: College of Information Technology, Jilin Agricultural University, Changchun 130118, China
Hanbo Liu: College of Information Technology, Jilin Agricultural University, Changchun 130118, China
Xudan Zhou: College of Forestry and Grassland, Jilin Agricultural University, Changchun 130118, China
Chunli Zhao: College of Forestry and Grassland, Jilin Agricultural University, Changchun 130118, China
Xiao Chen: College of Information Technology, Jilin Agricultural University, Changchun 130118, China
Lianxing Gao: College of Engineering and Technology, Jilin Agricultural University, Changchun 130118, China

Sustainability, 2023, vol. 15, issue 18, 1-23

Abstract: In order to address the limited scale and insufficient diversity of research datasets for maize leaf diseases, this study proposes a maize disease image generation algorithm based on the cycle generative adversarial network (CycleGAN). With the disease image transfer method, healthy maize images can be transformed into diseased crop images. To improve the accuracy of the generated data, the category activation mapping attention mechanism is integrated into the original CycleGAN generator and discriminator, and a feature recombination loss function is constructed in the discriminator. In addition, the minimum absolute error is used to calculate the differences between the hidden layer feature representations, and backpropagation is employed to enhance the contour information of the generated images. To demonstrate the effectiveness of this method, the improved CycleGAN algorithm is used to transform healthy maize leaf images. Evaluation metrics, such as peak signal-to-noise ratio (PSNR), structural similarity (SSIM), Fréchet inception distance (FID), and grayscale histogram can prove that the obtained maize leaf disease images perform better in terms of background and detail preservation. Furthermore, using this method, the original CycleGAN method, and the Pix2Pix method, the dataset is expanded, and a recognition network is used to perform classification tasks on different datasets. The dataset generated by this method achieves the best performance in the classification tasks, with an average accuracy rate of over 91 % . These experiments indicate the feasibility of this model in generating high-quality maize disease leaf images. It not only addresses the limitation of existing maize disease datasets but also improves the accuracy of maize disease recognition in small-sample maize leaf disease classification tasks.

Keywords: cycle-consistent adversarial networks; attention mechanism; maize leaf disease identification; feature recombination; computer vision (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://www.mdpi.com/2071-1050/15/18/13420/pdf (application/pdf)
https://www.mdpi.com/2071-1050/15/18/13420/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:15:y:2023:i:18:p:13420-:d:1235059

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().