DCS-YOLOv5s: A Lightweight Algorithm for Multi-Target Recognition of Potato Seed Potatoes Based on YOLOv5s

Abstract

The quality inspection of potato seed tubers is pivotal for their effective segregation and a critical step in the cultivation process of potatoes. Given the dearth of research on intelligent tuber-cutting machinery in China, particularly concerning the identification of bud eyes and defect detection, this study has developed a multi-target recognition approach for potato seed tubers utilizing deep learning techniques. By refining the YOLOv5s algorithm, a novel, lightweight model termed DCS-YOLOv5s has been introduced for the simultaneous identification of tuber buds and defects. This study initiates with data augmentation of the seed tuber images obtained via the image acquisition system, employing strategies such as translation, noise injection, luminance modulation, cropping, mirroring, and the Cutout technique to amplify the dataset and fortify the model’s resilience. Subsequently, the original YOLOv5s model undergoes a series of enhancements, including the substitution of the conventional convolutional modules in the backbone network with the depth-wise separable convolution DP_Conv module to curtail the model’s parameter count and computational load; the replacement of the original C3 module’s Bottleneck with the GhostBottleneck to render the model more compact; and the integration of the SimAM attention mechanism module to augment the model’s proficiency in capturing features of potato tuber buds and defects, culminating in the DCS-YOLOv5s lightweight model. The research findings indicate that the DCS-YOLOv5s model outperforms the YOLOv5s model in detection precision and velocity, exhibiting superior detection efficacy and model compactness. The model’s detection metrics, including Precision, Recall, and mean Average Precision at Intersection over Union thresholds of 0.5 (mAP1) and 0.75 (mAP2), have improved to 95.8%, 93.2%, 97.1%, and 66.2%, respectively, signifying increments of 4.2%, 5.7%, 5.4%, and 9.8%. The detection velocity has also been augmented by 12.07%, achieving a rate of 65 FPS. The DCS-YOLOv5s target detection model, by attaining model compactness, has substantially heightened the detection precision, presenting a beneficial reference for dynamic sample target detection in the context of potato-cutting machinery.