Automated detection of grapes and leaves in viticulture with a YOLOv2 neural network

Andreza Aparecida dos Santos, Sandra Eliza Fontes de Avila, Thiago Teixeira dos Santos
2018 Revista dos Trabalhos de Iniciação Científica da UNICAMP  
In this work, we modeled the problem of detection of fruit and leaves in viticulture for proximal applications as a supervised machine learning task. We created and manually labeled a database of images obtained at Guaspari Winery. In total, the database consists of 11.883 images of bunch of grapes and leaves. We trained a convolutional network with YOLOv2 architecture to locate and classify bunch of grapes and leaves. Quantitative tests have shown results for detection and classification with
more » ... lassification with precision of 100%, recall of 74,22% and F1-Score up to 85,2% for the class "grape". Also, qualitative tests show that the model generalizes well when tested on photographs of other grape varieties. These results are promising and are moving towards the possibility of application in the field.
doi:10.20396/revpibic262018155 fatcat:zcwi7knue5et5mmdlxypp4sdvi