YOLO-Peach: A High-Performance Lightweight YOLOv8s-Based Model for Accurate Recognition and Enumeration of Peach Seedling Fruits

Abstract

The identification and enumeration of peach seedling fruits are pivotal in the realm of precision agriculture, greatly influencing both yield estimation and agronomic practices. This study introduces an innovative, lightweight YOLOv8 model for the automatic detection and quantification of peach seedling fruits, designated as YOLO-Peach, to bolster the scientific rigor and operational efficiency of orchard management. Traditional identification methods, which are labor-intensive and error-prone, have been superseded by this advancement. A comprehensive dataset was meticulously curated, capturing the rich characteristics and diversity of peach seedling fruits through high-resolution imagery at various times and locations, followed by meticulous preprocessing to ensure data quality. The YOLOv8s model underwent a series of lightweight optimizations, including the integration of MobileNetV3 as its backbone, the p2BiFPN architecture, spatial and channel reconstruction convolution, and coordinate attention mechanism, all of which have significantly bolstered the model’s capability to detect small targets with precision. The YOLO-Peach model excels in detection accuracy, evidenced by a precision and recall of 0.979, along with an mAP50 of 0.993 and an mAP50-95 of 0.867, indicating its superior capability for peach sapling identification with efficient computational performance. The findings underscore the model’s efficacy and practicality in the context of peach seedling fruit recognition. Ablation studies have shed light on the indispensable role of each component, with MobileNetV3 streamlining the model’s complexity and computational load, while the p2BiFPN architecture, ScConv convolutions, and coordinate attention mechanism have collectively enhanced the model’s feature extraction and detection precision for minute targets. The implications of this research are profound, offering a novel approach to peach seedling fruit recognition and serving as a blueprint for the identification of young fruits in other fruit species. This work holds significant theoretical and practical value, propelling forward the broader field of agricultural automation.