Detection of small foreign objects in Pu-erh sun-dried green tea: An enhanced YOLOv8 neural network model based on deep learning

Abstract

To efficiently and accurately detect minuscule foreign objects in the processing of Pu-erh sun-dried green tea, ensuring food quality and consumer safety, this study innovatively proposes an enhanced YOLOv8 neural network based on deep learning. In light of the shortcomings of the traditional YOLOv8 network, to further enhance the model's ability to identify foreign object targets, improve the depth and breadth of feature extraction, and to better understand the contextual connections between different parts of the image, this study employs a Shape-IoU optimized loss function. It replaces parts of the network structure with Receptive-Field Attention Convolution technology and embeds Double Attention Networks to optimize the network. Experimental results show that the enhanced YOLOv8 neural network model achieves a precision rate of 98.35% for the detection of foreign objects in Pu-erh sun-dried green tea, which is a 3.93% increase compared to the original YOLOv8 network. Compared to mainstream detection models such as YOLOv7, YOLOv5, Faster-RCNN, CornerNet, and SSD, the mean Average Precision values of the enhanced YOLOv8 network model have significantly increased by 4.48%, 6.66%, 13.63%, 13.20%, and 9.84% respectively. This enhanced YOLOv8 network provides a viable research method and significant reference for the detection of small foreign objects in Pu-erh sun-dried green tea, holding substantial importance for tea-producing enterprises and food safety regulatory authorities. Furthermore, this study offers a more comprehensive and efficient solution for foreign object detection in Pu-erh tea and the broader food industry, laying the groundwork for the modernization and intelligentization of food safety and quality control.