Soybean Leaf Diseases Recognition Based on Generative Adversarial Network and Transfer Learning

Abstract

Soybean leaf disease labeling data are not easy to obtain, and soybean leaf disease model training often needs a lot of data. Due to the limitations of fixed rules such as rotation and clipping, traditional data enhancement cannot generate images with diversity and variability. In view of the above problems, this study proposed a data enhancement method based on generative adversarial network to expand the original soybean leaf disease dataset. This method was based on cyclic confrontation network, and its discriminator uses dense connection strategy to realize feature reuse, so as to reduce the amount of calculation. In the training process, improved transfer learning is used to automatically fine tune the pre-training model. The accuracy of the optimized method in 9 kinds of soybean leaf disease image recognition is 95.84%, which is 0.98% higher than the traditional fine-tuning method. The experimental results show that this method based on generating confrontation network has significant ability in generating soybean leaf disease image, and can expand the existing dataset. In addition, this method also provides an effective data enhancement solution for the expansion of other crop disease image datasets.