Application of lightweight YOLOv5 for walnut kernel grade classification and endogenous foreign body detection

Abstract

The appearance quality grading of walnut kernels is extremely important in the deep processing of walnuts. Foreign body pollution in food is also a challenging problem that has been troubling food companies. Therefore, it is necessary to propose an efficient and non-destructive detection method for walnut kernels and their endogenous foreign bodies. To improve the quality and safety of walnut deep-processed products, this paper proposes an improved detection model based on the original YOLO V5. The improved model achieves a reduction in memory consumption by 38% and computational workload by 34%. Additionally, it increases the average precision of the model by 1.1% and improves CPU speed by 52%. When compared to other lightweight networks, our model demonstrates optimal overall performance. Moreover, our algorithm significantly enhances the precision of detecting small targets and target selection, as compared to the original model.By reducing the input image size of the model, it is able to meet the requirements of real-time detection of walnut kernels and their endogenous foreign bodies on the edge device Raspberry Pi. This provides a technical reference for the non-destructive detection of food quality and its endogenous foreign bodies.