Acta Agriculturae Zhejiangensis ›› 2023, Vol. 35 ›› Issue (7): 1729-1739.DOI: 10.3969/j.issn.1004-1524.20221148

• Biosystems Engineering • Previous Articles     Next Articles

Identification of harm grades of cotton spider mites based on transfer learning and improved residual network

ZHANG Yana,b(), ZHOU Baopinga,*(), WANG Yua,b, FENG Jiea,b, YE Fankaia,b, HE Yunlonga,b   

  1. a. College of Information Engineering, Tarim University, Alar 843300, Xinjiang, China
    b. Key Laboratory of Modern Agricultural Engineering, Tarim University, Alar 843300, Xinjiang, China
  • Received:2022-08-03 Online:2023-07-25 Published:2023-08-17
  • Contact: ZHOU Baoping

Abstract:

In view of the low accuracy, long time consuming and high cost of traditional artificial diagnosis of cotton spider mites, a method for detection and classification of harm grade was proposed based on transfer learning and improved residual network. The cotton leaf images either healthy or with 3 harm grades of cotton spider mites were collected both in single background and natural environment to construct image dataset. First, the PlantVillage dataset was used to pretrain the model. Data augmentation was carried out to expand training samples. Then, based on the original ResNet50 network, an improved ResNet50 network was constructed by introducing focal loss functions, embedding attention mechanism modules in different network layers, and optimizing with the Dropout regularization. Finally, the performance of the improved ResNet50 network was compared with other models. It was shown that with the attention mechanism module introduced both in the deep and shallow layers, the momentum being 0.9, and the learning rate being 0.001, the improved ResNet50 network had the best classification effect, which was superior than the original ResNet50, VGG16, MobileNet, AlexNet and SENet models, with the average recognition accuracy of 97.8%.

Key words: cotton spider mite, hazard level, ResNet50 network, transfer learning, focal loss function

CLC Number: