YOLO-Banana-Seg: a lightweight and efficient model for rapid segmentation of banana bunches and stalks in complex orchards

In banana orchards, accurate segmentation of bunches and stalks is vital for automated harvesting and yield estimation. However, color similarity between green bananas and leaves, irregular shapes, and occlusion makes instance segmentation highly challenging. This study proposes an improved YOLOv11-based segmentation model, YOLO-Banana-Seg, for banana bunches and stalks. We compiled a multi-condition banana dataset encompassing varying illumination intensities and occlusion levels, while the proposed model integrates multi-scale feature fusion and adaptive attention mechanisms to significantly improve occluded target segmentation accuracy. The results show that YOLO-Banana-Seg achieves high segmentation accuracy while reducing parameters and computational complexity, ensuring its applicability in resource-limited scenarios. Experiments demonstrate 97.4% accuracy and 82.9% recall with 16.7% fewer parameters than YOLOv11n, balancing precision and efficiency. The model's high precision and efficiency directly could support the development of harvesting robots and yield estimation systems for agricultural applications.

Chen M, Chen Z, Luo L, Tang Y, Cheng J, Wei H, et al., 2024. Dynamic visual servo control methods for continuous operation of a fruit harvesting robot working throughout an orchard. Comput Electron Agric 219:108774. DOI: https://doi.org/10.1016/j.compag.2024.108774

Chen T, Zhang R, Zhu L, Zhang S, Li X, 2021. A method of fast segmentation for banana stalk exploited lightweight multi-feature fusion deep neural network. Machines 9:66. DOI: https://doi.org/10.3390/machines9030066

Chen Y, Zhang C, Chen B, Huang Y, Sun Y, Wang C, et al., 2024. Accurate leukocyte detection based on deformable-DETR and multi-level feature fusion for aiding diagnosis of blood diseases. Comput Biol Med 170:107917. DOI: https://doi.org/10.1016/j.compbiomed.2024.107917

Cheng J, Zhu Y, Zhao Y, Li T, Chen M, Sun Q, et al., 2024. Application of an improved U-net with image-to-image translation and transfer learning in peach orchard segmentation. Int J Appl Earth Obs Geoinf 130:103871. DOI: https://doi.org/10.1016/j.jag.2024.103871

Deng F, He Z, Fu L, Chen J, Li N, Chen W, et al., 2025. A new maturity recognition algorithm for Xinhui citrus based on improved YOLOv8. Front Plant Sci 16:1472230. DOI: https://doi.org/10.3389/fpls.2025.1472230

Fu L, Wu F, Zou X, Jiang Y, Lin J, Yang Z, et al., 2022a. Fast detection of banana bunches and stalks in the natural environment based on deep learning. Comput Electron Agric 194:106800. DOI: https://doi.org/10.1016/j.compag.2022.106800

Fu L, Yang Z, Wu F, Zou X, Lin J, Cao Y, et al., 2022b. YOLO-banana: a lightweight neural network for rapid detection of banana bunches and stalks in the natural environment. Agronomy 12:391. DOI: https://doi.org/10.3390/agronomy12020391

Girshick R, Donahue J, Darrell T, Malik J, 2014. Rich feature hierarchies for accurate object detection and semantic segmentation. Proc. IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), Columbus; pp. 580-587. DOI: https://doi.org/10.1109/CVPR.2014.81

He J, Duan J, Yang Z, Ou J, Ou X, Yu S, et al., 2023. Method for segmentation of banana crown based on improved DeepLabv3+. Agronomy 13:1838. DOI: https://doi.org/10.3390/agronomy13071838

Koirala A, Walsh KB, Wang Z, McCarthy C, 2019. Deep learning for real-time fruit detection and orchard fruit load estimation: benchmarking of ‘MangoYOLO’. Precis Agric 20:1107-35. DOI: https://doi.org/10.1007/s11119-019-09642-0

Li H, Huang J, Gu Z, He D, Huang J, Wang C, 2024. Positioning of mango picking point using an improved YOLOv8 architecture with object detection and instance segmentation. Biosyst Eng 247:202-20. DOI: https://doi.org/10.1016/j.biosystemseng.2024.09.015

Li L, Li K, He Z, Li H, Cui Y, 2025. Kiwifruit segmentation and identification of picking point on its stem in orchards. Comput Electron Agric 229:109748. DOI: https://doi.org/10.1016/j.compag.2024.109748

Li Y, Feng Q, Liu C, Xiong Z, Sun Y, Xie F, et al., 2023. MTA-YOLACT: multitask-aware network on fruit bunch identification for cherry tomato robotic harvesting. Eur J Agron 146:126812. DOI: https://doi.org/10.1016/j.eja.2023.126812

Li Z, Wang D, Zhu T, Tao Y, Ni C, 2024. Review of deep learning-based methods for non-destructive evaluation of agricultural products. Biosyst Eng 245:56-83. DOI: https://doi.org/10.1016/j.biosystemseng.2024.07.002

Liang X, Wei Z, Chen K, 2025. A method for segmentation and localization of tomato lateral pruning points in complex environments based on improved YOLOv5. Comput Electron Agric 229:109731. DOI: https://doi.org/10.1016/j.compag.2024.109731

Liu M, Chen W, Cheng J, Wang Y, Zhao C, 2024. Y-HRNet: research on multi-category cherry tomato instance segmentation model based on improved YOLOv7 and HRNet fusion. Comput Electron Agric 227:109531. DOI: https://doi.org/10.1016/j.compag.2024.109531

Liu Q, Lv J, Zhang C, 2024. MAE-YOLOv8-based small object detection of green crisp plum in real complex orchard environments. Comput Electron Agric 226:109458. DOI: https://doi.org/10.1016/j.compag.2024.109458

Lu Y, Ji Z, Yang L, Jia W, 2023. Mask positioner: an effective segmentation algorithm for green fruit DOI: https://doi.org/10.1016/j.jksuci.2023.101598

Peng H, Li Z, Zou X, Wang H, Xiong J, 2025. Research on litchi image detection in orchard using UAV based on improved YOLOv5. Expert Syst Appl 263:125828. DOI: https://doi.org/10.1016/j.eswa.2024.125828

Redmon J, Divvala S, Girshick R, Farhadi A, 2016. You only look once: unified, real-time object detection. Proc. IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), Las Vegas; pp. 779-788. DOI: https://doi.org/10.1109/CVPR.2016.91

Ren S, He K, Girshick R, Sun J, 2017. Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans Pattern Anal Mach Intell 39:1137-49. DOI: https://doi.org/10.1109/TPAMI.2016.2577031

in complex environment. J King Saud Univ Comput Inf Sci 35:101598.

Ryan M, 2023. Labour and Skills Shortages in the Agro-Food Sector. OECD Food, Agriculture and Fisheries Papers, No. 189. Paris, OECD Publishing.

Tan M, Pang R, Le QV, 2020. EfficientDet: scalable and efficient object detection. In: Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). Seattle, WA, USA, IEEE; pp. 10778-87. DOI: https://doi.org/10.1109/CVPR42600.2020.01079

Tang Y, Qi S, Zhu L, Zhuo X, Zhang Y, Meng F, 2024. Obstacle avoidance motion in mobile robotics. J Syst Simul 36:1.

Wan D, Lu R, Shen S, Xu T, Lang X, Ren Z, 2023. Mixed local channel attention for object detection. Eng Appl Artif Intell 123:106442. DOI: https://doi.org/10.1016/j.engappai.2023.106442

Wang C, Li C, Han Q, Wu F, Zou X, 2023. A performance analysis of a litchi picking robot system for actively removing obstructions, using an artificial intelligence algorithm. Agronomy 13:2795. DOI: https://doi.org/10.3390/agronomy13112795

Wang P, Niu T, He D, 2021. Tomato young fruits detection method under near color background based on improved Faster R-CNN with attention mechanism. Agriculture 11:1059. DOI: https://doi.org/10.3390/agriculture11111059

Wu F, Yang Z, Mo X, Wu Z, Tang W, Duan J, et al., 2023. Detection and counting of banana bunches by integrating deep learning and classic image-processing algorithms. Comput Electron Agric 209:107827. DOI: https://doi.org/10.1016/j.compag.2023.107827

Ye L, Wu F, Zou X, Li J, 2023. Path planning for mobile robots in unstructured orchard environments: an improved kinematically constrained bi-directional RRT approach. Comput Electron Agric 215:108453. DOI: https://doi.org/10.1016/j.compag.2023.108453

Yu Y, Zhang K, Yang L, Zhang D, 2019. Fruit detection for strawberry harvesting robot in non-structural environment based on Mask R-CNN. Comput Electron Agric 163:104846. DOI: https://doi.org/10.1016/j.compag.2019.06.001

Yue X, Qi K, Na X, Zhang Y, Liu Y, Liu C, 2023. Improved YOLOv8-seg network for instance segmentation of healthy and diseased tomato plants in the growth stage. Agriculture 13:1643. DOI: https://doi.org/10.3390/agriculture13081643

Zhou L, Yang Z, Deng F, Zhang J, Xiao Q, Fu L, et al., 2024. Banana bunch weight estimation and stalk central point localization in banana orchards based on RGB-D images. Agronomy 14:1123. DOI: https://doi.org/10.3390/agronomy14061123