RSWD-YOLO: A Walnut Detection Method Based on UAV Remote Sensing Images

Accurate walnut yield prediction is crucial for the development of the walnut industry. Traditional manual counting methods are limited by labor and time costs, leading to inaccurate walnut quantity assessments. In this paper, we propose a walnut detection method based on UAV (UAV means Unmanned Aerial Vehicle) remote sensing imagery to improve the walnut yield prediction accuracy. Based on the YOLOv11 network, we propose several improvements to enhance the multi-scale object detection capability while achieving a more lightweight model structure. Specifically, we reconstruct the feature fusion network with a hierarchical scale-based feature pyramid structure and implement lightweight improvements to the feature extraction component. These modifications result in the RSWD-YOLO network (RSWD means remote sensing walnut detection; YOLO means ‘You Only Look Once’, and it is the specific abbreviation used for a series of object detection algorithms), which is specifically designed for walnut detection. Furthermore, to optimize the detection performance under hardware resource constraints, we apply knowledge distillation to RSWD-YOLO, thereby further improving the detection accuracy. Through model deployment and testing on small edge devices, we demonstrate the feasibility of our proposed method. The detection algorithm achieves 86.1% mean Average Precision on the walnut dataset while maintaining operational functionality on small edge devices. The experimental results demonstrate that our proposed UAV remote sensing-based walnut detection method has a significant practical application value and can provide valuable insights for future research in related fields.