A Method for Quality Grading and Endogenous Impurity Detection of Walnut Kernels Based on Lightweight Deep Learning
This paper proposes a lightweight detection model YOLO-WQ based on improved YOLOv11n to meet the multiple requirements of high precision, lightweight, and real-time for quality grading and endogenous impurity detection in walnut kernel processing. Traditional methods are difficult to effectively identify small targets such as walnut kernels and their impurities with complex surface textures and similar shapes, while existing deep learning models suffer from high computational resource consumption and deployment difficulties. This study conducts triple optimization starting from the model structure: ShuffleNetV2 is used as the backbone network, significantly reducing the number of parameters and computational burden; Design a Slim neck neck structure that integrates GSConv and VoV-GSCSP to enhance multi-scale feature fusion capability; Introducing SEAM attention mechanism to improve the perception accuracy of walnut groove details and occluded areas. Experiments have shown that YOLO-WQ achieves success on self built datasets [email protected] At 90.6%, the accuracy and recall rates reached 84.8% and 87.4%, respectively, which were 2.5%, 4.2%, and 3.7% higher than the original model. The computational load (FLOPs) decreased by 44.4%, the model volume was reduced by 20%, and the inference speed reached 157.09 fps. Further testing on multiple publicly available agricultural product datasets has confirmed its excellent generalization performance. This model combines high precision and efficiency, and is suitable for real-time quality monitoring and impurity detection tasks in actual walnut processing production lines.
Authors
- Shaomin Lu
- Jiahao Liu
- Jianlong Li (ORCID: https://orcid.org/0000-0001-7845-6006)
- Wei Li (ORCID: https://orcid.org/0000-0001-6282-2761)
- Jianan Bai (ORCID: https://orcid.org/0009-0000-5614-0502)
- Jianchao Wang (ORCID: https://orcid.org/0000-0002-8488-1903)
- Hongsen Liao (ORCID: https://orcid.org/0009-0006-7980-9057)
- Hailong Ti (ORCID: https://orcid.org/0009-0002-2652-5254)
Institutions
- Southwest Forestry University (CN)
Publication Details
- Journal
- IEEE Access
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1109/access.2025.3626714
- Primary Topic
- Smart Agriculture and AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00