作者
Tianzuo Li,Jianxin Xue,Miaomiao Wei,Xinming Yuan,Xindong Wang,Zimeng Zhang
摘要
Accurate detection of fresh jujube fruits plays a vital role in precision agriculture, enabling reliable yield estimation and supporting automation tasks such as robotic harvesting. To address the challenges of detecting such small targets (≤32 × 32 pixels) in complex orchard environments, this study proposes JFST-DETR, an efficient and robust detection model based on the Real-Time DEtection TRansformer (RT-DETR). First, to address the insufficient feature representation for small jujube fruit targets, a novel module called the Global Awareness Adaptive Module (GAAM) is designed. Building on GAAM and the innovative Spatial Coding Module (SCM), a new Spatial Enhancement Pyramid Network (SEPN) is proposed. Through the spatial-depth transformation domain and global awareness adaptive processing units, SEPN captures fine-grained features of small targets, enhancing the detection accuracy for small objects. Second, a Dynamic Sampling (DySample) operator is adopted, which optimizes feature space details via dynamic offset calculation and lightweight design, improving detection accuracy while reducing computational costs. Finally, to solve the problem of complex background interference caused by foliage occlusion and illumination variations, Pinwheel-Shaped Convolution (PSConv) is introduced. By using asymmetric padding and multi-directional convolution, PSConv enhances the robustness of feature extraction, ensuring reliable recognition in complex agricultural environments. Experimental results show that JFST-DETR achieves precision, recall, F1, mAP@50, and mAP@50:95 of 93%, 86.8%, 89.8%, 94.3%, and 75.2%. Compared to the baseline model, these metrics improve by 0.8%, 3.7%, 2.4%, 2.6%, and 3.1%, respectively. Cross-dataset evaluations further confirm its strong generalizability, demonstrating potential as a practical solution for small-target detection in intelligent horticulture.