MDP-YOLO: A LIGHTWEIGHT YOLOV5S ALGORITHM FOR MULTI-SCALE PEST DETECTION

• Published in: Engenharia agrícola Vol. 43; no. 4
• Main Authors: Yu, Jianghua, Zhang, Bing
• Format: Journal Article
• Language: English
• Published: Sociedade Brasileira de Engenharia Agrícola 01.01.2023
• Subjects: attention mechanism; deep learning; lightweight; object detection; pest detection; YOLOv5s
• ISSN: 0100-6916; 1809-4430
• DOI: 10.1590/1809-4430-eng.agric.v43n4e20230065/2023

ABSTRACT Rice pest detection technology plays a crucial role in enabling food production, ensuring ecological balance, supporting sustainable agriculture, and promoting the health of farmers. However, existing technology is faced with challenges such as low detection accuracy, high computational complexity, and large model sizes, making it unsuitable for mobile deployment. This paper presents MDP-YOLO, a lightweight rice pest detection model based on the YOLOv5s model. It includes improvements such as the use of ShuffleNetV2 as the backbone network to significantly reduce the number of parameters and complexity of the model; the introduction of GhostConv to replace redundant convolutional layers, in order to further reduce the computational complexity and size; the integration of a large-scale feature extraction layer to enhance the ability of the algorithm to detect small rice pest objects; and the use of CBAM to increase the focus on the regions of interest. Tests on collected datasets show that the modified MDP-YOLO model increases the mean average precision by 5.5% and reduces the FLOPs, parameters, and model size by 72.4%, 89.2%, and 70.1%, respectively, compared to the original YOLOv5s model. Ablation experiments indicate that MDP-YOLO gives superior detection rice pest detection performance compared to other algorithms. MDP-YOLO can effectively detect rice pests, and provides theoretical and technical support for the deployment of lightweight rice pest detection models in practical scenarios.