关键词: 火龙果, 全生育期, YOLOv5s, WGAN-GP, MobileViT, 目标检测

Abstract: To address the challenges of traditional orchard management that heavily relied on manual labor and the difficulties in rapidly identifying pitaya fruits at different growth stages across large-scale orchards, this study categorized pitaya growth stages into seven distinct phases based on its developmental characteristics. Firstly, the dataset was augmented using a Generative Adversarial Network (WGAN-GP) with conditional discrimination mechanisms to enhance rare sample representation and improve dataset balance. Secondly, building upon the YOLOv5s object detection framework, the core architecture was replaced with the lightweight MobileViT network to maintain detection accuracy while significantly accelerating model inference speed. The experimental results demonstrated that the enhanced model with the optimized dataset and network architecture achieved a precision of 85.7% and a recall rate of 77.6% on the test set, which increased by 2.5 and 0.9 percentage points compared to the original model, respectively. The average detection time of a single image was 18.64 ms, which was 3.87 ms shorter than the YOLOv5s model, and the improved detection network achieved a mAP₅₀ value of 82.5%, a mAP_50-95 value of 62.4%, while the model size was 5.87×10⁶. This system could achieve real-time detection of pitaya in natural environments and provide a feasible visual inspection solution for subsequent monitoring of pitaya growth status and automated operations.

Key words: pitaya, whole life cycle, YOLOv5s, WGAN-GP, MobileViT, object detection