Back to Search Start Over

A Lightweight Crop Pest Classification Method Based on Improved MobileNet-V2 Model.

Authors :
Peng, Hongxing
Xu, Huiming
Shen, Guanjia
Liu, Huanai
Guan, Xianlu
Li, Minhui
Source :
Agronomy. Jun2024, Vol. 14 Issue 6, p1334. 20p.
Publication Year :
2024

Abstract

This paper proposes PestNet, a lightweight method for classifying crop pests, which improves upon MobileNet-V2 to address the high model complexity and low classification accuracy commonly found in pest classification research. Firstly, the training phase employs the AdamW optimizer and mixup data augmentation techniques to enhance the model's convergence and generalization capabilities. Secondly, the Adaptive Spatial Group-Wise Enhanced (ASGE) attention mechanism is introduced and integrated into the inverted residual blocks of the MobileNet-V2 model, boosting the model's ability to extract both local and global pest information. Additionally, a dual-branch feature fusion module is developed using convolutional kernels of varying sizes to enhance classification performance for pests of different scales under real-world conditions. Lastly, the model's activation function and overall architecture are optimized to reduce complexity. Experimental results on a proprietary pest dataset show that PestNet achieves classification accuracy and an F1 score of 87.62% and 86.90%, respectively, marking improvements of 4.20 percentage points and 5.86 percentage points over the baseline model. Moreover, PestNet's parameter count and floating-point operations are reduced by 14.10% and 37.50%, respectively, compared to the baseline model. When compared with ResNet-50, MobileNet V3-Large, and EfficientNet-B1, PestNet offers superior parameter efficiency and floating-point operation requirements, as well as improved pest classification accuracy. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20734395
Volume :
14
Issue :
6
Database :
Academic Search Index
Journal :
Agronomy
Publication Type :
Academic Journal
Accession number :
178159167
Full Text :
https://doi.org/10.3390/agronomy14061334