A Feature Enhancement Framework for Joint Mango Fruit and Stem Detection in Complex Orchard Environments

Abstract

Reliable joint detection of mango fruits and stems is an essential upstream perception task for robotic harvesting, but remains challenging because stems are small, slender, frequently occluded, and visually degraded by illumination variation. This study proposes MangoNET, a YOLOv11n-based framework for joint mango fruit and stem detection in complex orchard environments. A P2 high-resolution detection head preserves fine spatial information for small targets, while SPPF-ELAN aggregates local and contextual features for partially visible objects. SENet recalibrates channel responses under illumination variation, and WIoU v3 regulates bounding-box samples with different localization qualities. A dataset containing 1782 original images of Tainong and Jinhuang mangoes was collected from two orchards and data augmentation was applied only to the training set, increasing its size from 1172 to 2886 images through rotation, contrast adjustment, and Gaussian noise addition. MangoNET achieved fruit and stem F1-scores of 0.920 and 0.916, respectively, with mAP50 and mAP50–95 values of 0.941 and 0.690. Compared with YOLOv11n, mAP50 and mAP50–95 increased by 1.6 and 2.9 percentage points, respectively, while stem recall increased from 0.877 to 0.906. Source-image-independent five-fold cross-validation yielded mean mAP50 and mAP50–95 values of 0.944 and 0.711, respectively. Pilot evaluations using images acquired by a UAV and an RGB-D camera in a geographically distinct orchard suggested that MangoNET could maintain detection performance in a different orchard environment. MangoNET supplies fruit and stem candidate regions for subsequent association, harvesting-point localization, and robotic manipulation.