The application discloses a
tea leaf recognition method based on knowledge
distillation and improved
pruning technology, and comprises
image acquisition and preprocessing,
tea leaf images are acquired through an
image acquisition device, color interference is eliminated through graying,
horizontal and vertical gradients are calculated using a
Sobel operator, low-quality images are screened through gradient amplitude variance, a threshold value M is set, images with a variance less than M are removed, input
data quality is improved, a
data set is constructed and enhanced, manual labeling is performed using a labeling tool, a
data set is generated, the generated
data set is given to an improved RT-DETR model for training, and a high-precision
tea leaf picking teacher model is obtained; the application effectively compensates for the precision loss caused by
pruning, makes the model perform excellently in the precision and speed trade-off, effectively compensates for the precision loss caused by
pruning, performs excellently in the precision and speed trade-off, is superior to other pruning and
distillation combination methods, and adopts mixed pruning of LAMP and channel pruning and multi-layer feature
distillation based on a Mimic loss.