Next Article in Journal
Genome-Wide Identification and Characterization of the PPPDE Gene Family in Rice
Next Article in Special Issue
Inversion of Glycyrrhiza Chlorophyll Content Based on Hyperspectral Imagery
Previous Article in Journal
The Valorization of Spanish Minority Grapevine Varieties—The Volatile Profile of Their Wines as a Characterization Feature
Previous Article in Special Issue
Detection and Analysis of Chili Pepper Root Rot by Hyperspectral Imaging Technology
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Improved YOLOv8 and SAHI Model for the Collaborative Detection of Small Targets at the Micro Scale: A Case Study of Pest Detection in Tea

1
College of Food Science and Technology, Yunnan Agricultural University, Kunming 650201, China
2
Big Data College, Yunnan Agricultural University, Kunming 650201, China
3
The Key Laboratory for Crop Production and Smart Agriculture of Yunnan Province, Kunming 650201, China
*
Author to whom correspondence should be addressed.
Agronomy 2024, 14(5), 1034; https://doi.org/10.3390/agronomy14051034
Submission received: 22 April 2024 / Revised: 7 May 2024 / Accepted: 8 May 2024 / Published: 13 May 2024

Abstract

Pest target identification in agricultural production environments is challenging due to the dense distribution, small size, and high density of pests. Additionally, changeable environmental lighting and complex backgrounds further complicate the detection process. This study focuses on enhancing the recognition performance of tea pests by introducing a lightweight pest image recognition model based on the improved YOLOv8 architecture. First, slicing-aided fine-tuning and slicing-aided hyper inference (SAHI) are proposed to partition input images for enhanced model performance on low-resolution images and small-target detection. Then, based on an ELAN, a generalized efficient layer aggregation network (GELAN) is designed to replace the C2f module in the backbone network, enhance its feature extraction ability, and construct a lightweight model. Additionally, the MS structure is integrated into the neck network of YOLOv8 for feature fusion, enhancing the extraction of fine-grained and coarse-grained semantic information. Furthermore, the BiFormer attention mechanism, based on the Transformer architecture, is introduced to amplify target characteristics of tea pests. Finally, the inner-MPDIoU, based on auxiliary borders, is utilized as a replacement for the original loss function to enhance its learning capacity for complex pest samples. Our experimental results demonstrate that the enhanced YOLOv8 model achieves a precision of 96.32% and a recall of 97.95%, surpassing those of the original YOLOv8 model. Moreover, it attains an mAP@50 score of 98.17%. Compared to Faster R-CNN, SSD, YOLOv5, YOLOv7, and YOLOv8, its average accuracy is 17.04, 11.23, 5.78, 3.75, and 2.71 percentage points higher, respectively. The overall performance of YOLOv8 outperforms that of current mainstream detection models, with a detection speed of 95 FPS. This model effectively balances lightweight design with high accuracy and speed in detecting small targets such as tea pests. It can serve as a valuable reference for the identification and classification of various insect pests in tea gardens within complex production environments, effectively addressing practical application needs and offering guidance for the future monitoring and scientific control of tea insect pests.
Keywords: small object detection; BiFormer; YOLOv8; SAHI; GELAN; tea pest damage small object detection; BiFormer; YOLOv8; SAHI; GELAN; tea pest damage

Share and Cite

MDPI and ACS Style

Ye, R.; Gao, Q.; Qian, Y.; Sun, J.; Li, T. Improved YOLOv8 and SAHI Model for the Collaborative Detection of Small Targets at the Micro Scale: A Case Study of Pest Detection in Tea. Agronomy 2024, 14, 1034. https://doi.org/10.3390/agronomy14051034

AMA Style

Ye R, Gao Q, Qian Y, Sun J, Li T. Improved YOLOv8 and SAHI Model for the Collaborative Detection of Small Targets at the Micro Scale: A Case Study of Pest Detection in Tea. Agronomy. 2024; 14(5):1034. https://doi.org/10.3390/agronomy14051034

Chicago/Turabian Style

Ye, Rong, Quan Gao, Ye Qian, Jihong Sun, and Tong Li. 2024. "Improved YOLOv8 and SAHI Model for the Collaborative Detection of Small Targets at the Micro Scale: A Case Study of Pest Detection in Tea" Agronomy 14, no. 5: 1034. https://doi.org/10.3390/agronomy14051034

APA Style

Ye, R., Gao, Q., Qian, Y., Sun, J., & Li, T. (2024). Improved YOLOv8 and SAHI Model for the Collaborative Detection of Small Targets at the Micro Scale: A Case Study of Pest Detection in Tea. Agronomy, 14(5), 1034. https://doi.org/10.3390/agronomy14051034

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop