Affiliation:
1. College of Engineering, Northeast Agricultural University, Harbin 150030, China
2. College of Arts and Sciences, Northeast Agricultural University, Harbin 150030, China
3. College of Electrical and Information, Northeast Agricultural University, Harbin 150030, China
Abstract
With the application of computer vision in the field of agricultural disease recognition, the convolutional neural network is widely used in grape leaf disease recognition and has achieved remarkable results. However, most of the grape leaf disease recognition models have the problem of weak generalization ability. In order to overcome this challenge, this paper proposes an image identification method for grape leaf diseases in different domains based on Fourier domain adaptation. Firstly, Fourier domain adaptation is performed on the labeled source domain data and the unlabeled target domain data. To decrease the gap in distribution between the source domain data and the target domain data, the low-frequency spectrum of the source domain data and the target domain data is swapped. Then, three convolutional neural networks (AlexNet, VGG13, and ResNet101) were used to train the images after style changes and the unlabeled target domain images were classified. The highest accuracy of the three networks can reach 94.6%, 96.7%, and 91.8%, respectively, higher than that of the model without Fourier transform image training. In order to reduce the impact of randomness, when selecting the transformed image, we propose using farthest point sampling to select the image with low feature correlation for the Fourier transform. The final identification result is also higher than the accuracy of the network model trained without transformation. Experimental results showed that Fourier domain adaptation can improve the generalization ability of the model and obtain a more accurate grape leaf disease recognition model.
Funder
The National Social Science Foundation of China