LBS-YOLO: a lightweight model for strawberry ripeness detection
Haitao Fu1, Xueying Li1, Zheng Li1
1College of Information Technology, Jilin Agricultural University, Changchun, China.
Introduction:
The traditional strawberry picking operation has long relied on manual work. With the aging trend of the population becoming more and more obvious, the application of intelligent picking technology has become an irreversible trend. However, existing recognition methods still face bottlenecks such as suboptimal recognition accuracy and low computational efficiency. To address these issues, this study constructs a lightweight detection model, LBS-YOLO, based on an improved YOLOv11n architecture, significantly the model's accuracy and interference robustness while greatly compressing the parameter quantity.
Methods:
The LBS-YOLO model is built upon YOLOv11n as the baseline network. In order to enhance the ability of backbone network feature representation, the model designs a lightweight LAWDS module. This design combines channel attention with spatial reconstruction operation to optimize the information retention efficiency in the down-sampling process, thus effectively enhancing the multi-scale feature representation ability and gradient flow propagation performance. Then in the feature fusion stage, the model introduces a Bidirectional Feature Pyramid Network (BiFPN), which not only enables cross-scale feature fusion but also achieves adaptive weighting through a learnable weight allocation mechanism. At last, adopts the C3k2_Star module to replace the conventional C3K2 for improved feature representation.
Results:
On the used strawberry dataset, the LBS-YOLO model reached 88.6% mAP@0.5 and 75.8% mAP@0.5:0.95, which were 2.2 and 1.3 percentage points higher than YOLOv11n, respectively. The LBS-YOLO model improves the recall rate from 83.2% of YOLOv11n to 86.4%, and the F1-score from 81.2% to 82.9%. Its computational complexity is 6.6 GFLOPs and its reasoning speed is 260.7 FPS. Even better, LBS-YOLO only needs 3.4MB of storage space and 1.6 million parameters, which are 34.6% and 38% less than YOLOv11n respectively.
Discussion:
The experiment demonstrates that, the LBS-YOLO model can significantly reduce the number of parameters and effectively improve the detection accuracy and operation efficiency. It successfully alleviated the problems of false detection and missed detection, thereby providing reliable technical support for strawberry growth monitoring, maturity identification and automatic picking.
