Applied Sciences (May 2024)

Citrus Pest Identification Model Based on Improved ShuffleNet

  • Yan-Nan Yu,
  • Chun-Lin Xiong,
  • Ji-Chi Yan,
  • Yong-Bin Mo,
  • Shi-Qing Dou,
  • Zuo-Hua Wu,
  • Rong-Feng Yang

DOI
https://doi.org/10.3390/app14114437
Journal volume & issue
Vol. 14, no. 11
p. 4437

Abstract

Read online

To address the current issues of complex structures and low accuracies in citrus pest identification models, a lightweight pest identification model was proposed. First, a parameterized linear rectification function was introduced to avoid neuronal death. Second, the model’s attention to pest characteristics was improved by incorporating an improved mixed attention mechanism. Subsequently, the network structure of the original model was adjusted to reduce architectural complexity. Finally, by employing transfer learning, an SCHNet model was developed. The experimental results indicated that the proposed model achieved an accuracy rate of 94.48% with a compact size of 3.84 MB. Compared to the original ShuffleNet V2 network, the SCHNet model showed a 3.12% accuracy improvement while reducing the model size by 22.7%. The SCHNet model exhibited an excellent classification performance for citrus pest identification, enabling the accurate identification of citrus pests.

Keywords