A Lightweight Tea Pest Detection Algorithm Based on Improved YOLOv8: YOLO-SEM

Rong Ye,Guoqi Shao,Tong Li
DOI: https://doi.org/10.1145/3662739.3664741
2024-01-01
Abstract:In response to the challenges of low detection accuracy, false positives, false negatives, and lengthy processing times in tea pest detection, this study proposes an enhanced YOLOv8 method. The approach involves the integration of lightweight convolution (SEConv) to replace standard convolution, reducing network parameters and enhancing detection speed. To address multi-scale target detection, the SEF module is introduced as an enhancement module built on SEConv. Additionally, an efficient multi-scale attention EMA module is incorporated to emphasize target features and suppress background interference. Furthermore, a lightweight detection head is designed based on the multiscale convolutional attention module (MCA) to enhance detection accuracy. To tackle gradient issues in low-quality instances within the training dataset, the Wise-IoU bounding box loss function is introduced. Experimental results demonstrate that YOLOv8-SEM outperforms the YOLOv8s baseline model, improving precision, recall, and mAP@0.5 by 2.4%, 5.2%, and 2.9% respectively. The enhanced model exhibits improved robustness and higher accuracy, showcasing strong generalization abilities and utility in multi-scale pest detection tasks amidst complex backgrounds. Its significance for tea pest detection is noteworthy.
What problem does this paper attempt to address?