Identification of winter wheat pests and diseases based on improved convolutional neural network

Abstract

Abstract Wheat pests and diseases are one of the main factors affecting wheat yield. According to the characteristics of four common pests and diseases, an identification method based on improved convolution neural network is proposed. VGGNet16 is selected as the basic network model, but the problem of small dataset size is common in specific fields such as smart agriculture, which limits the research and application of artificial intelligence methods based on deep learning technology in the field. Data expansion and transfer learning technology are introduced to improve the training mode, and then attention mechanism is introduced for further improvement. The experimental results show that the transfer learning scheme of fine-tuning source model is better than that of freezing source model, and the VGGNet16 based on fine-tuning all layers has the best recognition effect, with an accuracy of 96.02%. The CBAM-VGGNet16 and NLCBAM-VGGNet16 models are designed and implemented. The experimental results show that the recognition accuracy of the test set of CBAM-VGGNet16 and NLCBAM-VGGNet16 is higher than that of VGGNet16. The recognition accuracy of CBAM-VGGNet16 and NLCBAM-VGGNet16 is 96.60 and 97.57%, respectively, achieving high precision recognition of common pests and diseases of winter wheat.