IMPROVED YOLOV8-ALGORITHM FOR SORTING FRESH WHITE TEA: COMBINING FEATURE ENHANCEMENT AND ATTENTION MECHANISM
改进的 YOLOV8 新鲜白茶分选算法:将特征增强与注意力相结合
DOI : https://doi.org/10.35633/inmateh-75-80
Authors
Abstract
In this paper, an improved intelligent sorting algorithm for YOLOv8 white tea fresh leaves is proposed to solve the problems of unclear tea grades and uneven product levels caused by mechanical picking. The algorithm introduces the Dynamic Snake Convolution module (DSConv) for feature enhancement and adds an attention mechanism module, the Multi-Head Self-Attention mechanism (MHSA). Experiments show that the YOLOv8-DsConv-MHSA algorithm has an average accuracy of 96.4% and an average detection rate of 126.6 FPS per second, which is the best algorithm for white tea fresh leaf sorting in the comprehensive comparison. After deploying the proposed YOLOv8-DsConv-MHSA algorithm onto the developed tea sorting machine and conducting experimental comparisons with existing tea sorting machines, it is evident that the screening rate has been enhanced by 10.7%, and the operational efficiency has increased by 20%.
Abstract in Chinese