SAF-YOLO: Super-resolution augmented detection model with visual state space enhancement for safflower filament picking
摘要
Cluttered backgrounds, variable shooting angles, and fluctuating lighting frequently induce missed or false detections of filaments by picking robots, especially for small and imbalanced targets in safflower fields. To address these inherent limitations, we propose SAF-YOLO, a novel detector specifically optimized for robotic safflower harvesting. The architecture integrates three complementary innovations to ensure robust perception in unstructured environments. Firstly, a Visual State Space Model (VSSM)-based VSS-SPPF module is integrated into the backbone to capture global spatial context, which effectively separates filaments from complex background clutter. Secondly, an Asymptotic Feature Pyramid Network (AFPN) adaptively merges multi-scale features to mitigate the scale discrepancies induced by variable shooting angles. Finally, a Super-Resolution Self-Supervised (SRSS) auxiliary branch regularizes backbone learning via reconstruction tasks, driving the model to learn illumination-invariant features that resist lighting fluctuations; this branch operates only during training and is removed at inference to preserve efficiency. Experimental results demonstrate SAF-YOLO achieves 90.1% Precision, 85.9% Recall, and 93.3%