GLS-YOLO: A Lightweight Tea Bud Detection Model in Complex Scenarios

Abstract

The efficiency of tea bud harvesting has been greatly enhanced, and human labor intensity significantly reduced, through the mechanization and intelligent management of tea plantations. A key challenge for harvesting machinery is ensuring both the freshness of tea buds and the integrity of the tea plants. However, achieving precise harvesting requires complex computational models, which can limit practical deployment. To address the demand for high-precision yet lightweight tea bud detection, this study proposes the GLS-YOLO detection model, based on YOLOv8. The model leverages GhostNetV2 as its backbone network, replacing standard convolutions with depthwise separable convolutions, resulting in substantial reductions in computational load and memory consumption. Additionally, the C2f-LC module is integrated into the improved model, combining cross-covariance fusion with a lightweight contextual attention mechanism to enhance feature recognition and extraction quality. To tackle the challenges posed by varying poses and occlusions of tea buds, Shape-IoU was employed as the loss function to improve the scoring of similarly shaped objects, reducing false positives and false negatives while improving the detection of non-rectangular or irregularly shaped objects. Experimental results demonstrate the model’s superior performance, achieving an [email protected] of 90.55%. Compared to the original YOLOv8, the model size was reduced by 38.85%, and the number of parameters decreased by 39.95%. This study presents innovative advances in agricultural robotics by significantly improving the accuracy and efficiency of tea bud harvesting, simplifying the configuration process for harvesting systems, and effectively lowering the technological barriers for real-world applications.