A Method for Real-Time Recognition of Safflower Filaments in Unstructured Environments Using the YOLO-SaFi Model

• Published in: Sensors (Basel, Switzerland) Vol. 24; no. 13; p. 4410
• Main Authors: Chen, Bangbang, Ding, Feng, Ma, Baojian, Wang, Liqiang, Ning, Shanping
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 08.07.2024
• Subjects: Accuracy; Algorithms; Datasets; Deep learning; lightweight; safflower filament; target detection; unstructured environment; YOLOv8; China; target detection; unstructured environment; safflower filament; YOLOv8; lightweight
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s24134410

The identification of safflower filament targets and the precise localization of picking points are fundamental prerequisites for achieving automated filament retrieval. In light of challenges such as severe occlusion of targets, low recognition accuracy, and the considerable size of models in unstructured environments, this paper introduces a novel lightweight YOLO-SaFi model. The architectural design of this model features a Backbone layer incorporating the StarNet network; a Neck layer introducing a novel ELC convolution module to refine the C2f module; and a Head layer implementing a new lightweight shared convolution detection head, Detect_EL. Furthermore, the loss function is enhanced by upgrading CIoU to PIoUv2. These enhancements significantly augment the model's capability to perceive spatial information and facilitate multi-feature fusion, consequently enhancing detection performance and rendering the model more lightweight. Performance evaluations conducted via comparative experiments with the baseline model reveal that YOLO-SaFi achieved a reduction of parameters, computational load, and weight files by 50.0%, 40.7%, and 48.2%, respectively, compared to the YOLOv8 baseline model. Moreover, YOLO-SaFi demonstrated improvements in recall, mean average precision, and detection speed by 1.9%, 0.3%, and 88.4 frames per second, respectively. Finally, the deployment of the YOLO-SaFi model on the Jetson Orin Nano device corroborates the superior performance of the enhanced model, thereby establishing a robust visual detection framework for the advancement of intelligent safflower filament retrieval robots in unstructured environments.