# Tea bud pose estimation and grading detection network based on improved YOLOv7

## TL;DR

A new deep learning model, YOLO-PC, improves tea bud detection and grading by estimating pose and classifying buds more accurately than previous methods.

## Contribution

The novel YOLO-PC model integrates dynamic snake convolution and attention mechanisms for enhanced tea bud pose estimation and classification.

## Key findings

- YOLO-PC achieves 91.5% accuracy for one-bud-one-leaf and 93.2% for one-bud-two-leaf scenarios.
- The model improves mean average precision by 7.26% and pose accuracy by 9.65% compared to YOLOv7-pose.
- YOLO-PC reduces model parameters by 14.99 M while maintaining high detection accuracy.

## Abstract

Intelligent recognition and rapid grading of tea buds are crucial for advancing tea-picking machinery; however, complex plantation backgrounds and inconsistent bud growth have limited traditional algorithms to merely identifying picking points, neglecting bud pose and grade, which restricts harvesting efficiency. To address these challenges, we propose YOLO-PC, a deep neural network designed for simultaneous tea bud pose estimation and classification, which incorporates a dynamic snake convolution (DSConv) module for enhanced shape feature extraction, an ELASPP-CSPC attention mechanism for improved spatial pooling, and EIoU loss to accelerate regression and boost localization accuracy. Experimental results demonstrate that the model achieves detection accuracies of 91.5% for one-bud-one-leaf and 93.2% for one-bud-two-leaf scenarios, with an average keypoint detection accuracy (Pose_mAP) of 89.7% and a Normalized Mean Error (NME) of 0.047; furthermore, compared to YOLOv7-pose, it increases mean average precision by 7.26% and pose accuracy by 9.65% while reducing parameters by 14.99 M. Ablation studies confirm the superior performance of the proposed model in tea bud detection, indicating its potential to provide robust practical support for adaptive and intelligent tea harvesting systems.