Back to Search Start Over

PestLite: A Novel YOLO-Based Deep Learning Technique for Crop Pest Detection

Authors :
Qing Dong
Lina Sun
Tianxin Han
Minqi Cai
Ce Gao
Source :
Agriculture, Vol 14, Iss 2, p 228 (2024)
Publication Year :
2024
Publisher :
MDPI AG, 2024.

Abstract

Timely and effective pest detection is essential for agricultural production, facing challenges such as complex backgrounds and a vast number of parameters. Seeking solutions has become a pressing matter. This paper, based on the YOLOv5 algorithm, developed the PestLite model. The model surpasses previous spatial pooling methods with our uniquely designed Multi-Level Spatial Pyramid Pooling (MTSPPF). Using a lightweight unit, it integrates convolution, normalization, and activation operations. It excels in capturing multi-scale features, ensuring rich extraction of key information at various scales. Notably, MTSPPF not only enhances detection accuracy but also reduces the parameter size, making it ideal for lightweight pest detection models. Additionally, we introduced the Involution and Efficient Channel Attention (ECA) attention mechanisms to enhance contextual understanding. We also replaced traditional upsampling with Content-Aware ReAssembly of FEatures (CARAFE), which enable the model to achieve higher mean average precision in detection. Testing on a pest dataset showed improved accuracy while reducing parameter size. The mAP50 increased from 87.9% to 90.7%, and the parameter count decreased from 7.03 M to 6.09 M. We further validated the PestLite model using the IP102 dataset, and on the other hand, we conducted comparisons with mainstream models. Furthermore, we visualized the detection targets. The results indicate that the PestLite model provides an effective solution for real-time target detection in agricultural pests.

Details

Language :
English
ISSN :
20770472
Volume :
14
Issue :
2
Database :
Directory of Open Access Journals
Journal :
Agriculture
Publication Type :
Academic Journal
Accession number :
edsdoj.26745838ed3d44dbb458084c8370c71c
Document Type :
article
Full Text :
https://doi.org/10.3390/agriculture14020228