1. Simultaneous fruit detection and size estimation using multitask deep neural networks.
- Author
-
Ferrer-Ferrer, Mar, Ruiz-Hidalgo, Javier, Gregorio, Eduard, Vilaplana, Verónica, Morros, Josep-Ramon, and Gené-Mola, Jordi
- Subjects
- *
ARTIFICIAL neural networks , *FRUIT , *ARCHITECTURAL design , *APPLES - Abstract
The measurement of fruit size is of great interest to estimate the yield and predict the harvest resources in advance. This work proposes a novel technique for in-field apple detection and measurement based on Deep Neural Networks. The proposed framework was trained with RGB-D data and consists of an end-to-end multitask Deep Neural Network architecture specifically designed to perform the following tasks: 1) detection and segmentation of each fruit from its surroundings; 2) estimation of the diameter of each detected fruit. The methodology was tested with a total of 15,335 annotated apples at different growth stages, with diameters varying from 27 mm to 95 mm. Fruit detection results reported an F1-score for apple detection of 0.88 and a mean absolute error of diameter estimation of 5.64 mm. These are state-of-the-art results with the additional advantages of: a) using an end-to-end multitask trainable network; b) an efficient and fast inference speed; and c) being based on RGB-D data which can be acquired with affordable depth cameras. On the contrary, the main disadvantage is the need of annotating a large amount of data with fruit masks and diameter ground truth to train the model. Finally, a fruit visibility analysis showed an improvement in the prediction when limiting the measurement to apples above 65% of visibility (mean absolute error of 5.09 mm). This suggests that future works should develop a method for automatically identifying the most visible apples and discard the prediction of highly occluded fruits. • An end-to-end trainable multi-task deep neural network was designed. • The network includes two branches: instance segmentation and size regression. • The architecture was adapted to be used with 4-channel RGB + D images. • Fruit detection results reported an F1-score of 0.88. • Fruit size estimation results reported a mean absolute error of 5.64 mm. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF