A grain flow detection method based on an improved YOLOv8n model and dual-line counting with a multi-criterion discrimination mechanism

Abstract

To support non-contact grain yield monitoring in smart agriculture, this study presents a deep learning-based method for grain flow detection. The system includes two core components: an enhanced YOLOv8n detection model and a dual-line counting algorithm guided by multiple decision rules. In real harvesting conditions, challenges such as occlusion, fast grain movement, poor lighting, and small target sizes often lead to missed detections when using standard YOLO models. To improve performance, the YOLOv8n structure is modified in three ways. A minimum point distance IoU loss (MPDIoU) is applied to better distinguish overlapping grains. A spatial-channel shared attention (SCSA) module is added to help the model focus on weak or small grain targets. A lightweight Spatial and Channel reconstruction Convolution (SCConv) module is inserted into the network to reduce computation while maintaining accuracy. As a result, the updated model reaches 90.6 % in precision, 91.4 % in recall, and 94.5 % in mAP, improving the original version by 2.7, 1.7, and 2.5 points. In the counting stage, a dual-line method is used, combining rules such as movement direction checking, two-line crossing validation, abnormal shape removal, and IoU-based filtering. The full system remains stable even when grains move quickly, supporting reliable detection under field conditions.