Version 1
: Received: 3 September 2024 / Approved: 4 September 2024 / Online: 4 September 2024 (08:10:37 CEST)
How to cite:
Oliveira, F.; da Silva, D. Q.; Filipe, V.; Pinho, T. M.; Cunha, M.; Cunha, J. B.; Santos, F. N. Enhancing Grapevine Node Detection to Support Pruning Automation: Leveraging State-of-the-Art YOLO Detection Models for 2D Image Analysis. Preprints2024, 2024090322. https://doi.org/10.20944/preprints202409.0322.v1
Oliveira, F.; da Silva, D. Q.; Filipe, V.; Pinho, T. M.; Cunha, M.; Cunha, J. B.; Santos, F. N. Enhancing Grapevine Node Detection to Support Pruning Automation: Leveraging State-of-the-Art YOLO Detection Models for 2D Image Analysis. Preprints 2024, 2024090322. https://doi.org/10.20944/preprints202409.0322.v1
Oliveira, F.; da Silva, D. Q.; Filipe, V.; Pinho, T. M.; Cunha, M.; Cunha, J. B.; Santos, F. N. Enhancing Grapevine Node Detection to Support Pruning Automation: Leveraging State-of-the-Art YOLO Detection Models for 2D Image Analysis. Preprints2024, 2024090322. https://doi.org/10.20944/preprints202409.0322.v1
APA Style
Oliveira, F., da Silva, D. Q., Filipe, V., Pinho, T. M., Cunha, M., Cunha, J. B., & Santos, F. N. (2024). Enhancing Grapevine Node Detection to Support Pruning Automation: Leveraging State-of-the-Art YOLO Detection Models for 2D Image Analysis. Preprints. https://doi.org/10.20944/preprints202409.0322.v1
Chicago/Turabian Style
Oliveira, F., José Boaventura Cunha and Filipe N. Santos. 2024 "Enhancing Grapevine Node Detection to Support Pruning Automation: Leveraging State-of-the-Art YOLO Detection Models for 2D Image Analysis" Preprints. https://doi.org/10.20944/preprints202409.0322.v1
Abstract
Automating pruning tasks entails overcoming several challenges, encompassing not only robotic manipulation but also environment perception and detection. To achieve efficient pruning, robotic systems must accurately identify the correct cutting points. A possible method to define these points is to choose the cutting location based on the number of nodes present on the targeted cane. For this purpose, in grapevine pruning, it is required to correctly identify the nodes present on the primary canes of the grapevines. In this paper, a novel method of node detection in grapevines is proposed with four distinct state-of-the-art versions of the YOLO detection model: YOLOv7, YOLOv8, YOLOv9 and YOLOv10. These models were trained on a public dataset with images containing artificial backgrounds and afterwards validated on different cultivars of grapevines from two distinct Portuguese viticulture regions with cluttered backgrounds. This allowed to evaluate the robustness of the algorithms on the detection of nodes in diverse environments, compare the performance of the YOLO models used, as well as create a publicly available dataset of grapevines obtained in Portuguese vineyards for node detection. Overall, all used models were capable of achieving correct node detection on images of grapevines from the three distinct datasets. Considering the trade-off between accuracy and inference speed, the YOLOv7 model demonstrated to be the most robust in detecting nodes in 2D images of grapevines, achieving F1-Score values between 70 % and 86.5 % with inference times of around 89 ms for an input size of 1280×1280 px. Considering these results, this work contributes with an efficient approach for real-time node detection for further implementation on an autonomous robotic pruning system.
Keywords
deep learning; precision agriculture; pruning; robotic systems; YOLO
Subject
Computer Science and Mathematics, Artificial Intelligence and Machine Learning
Copyright:
This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.