Using pre-trained convolutional neural networks to predict maturity levels of strawberries
J.R. Buitenweg (TU Delft - Electrical Engineering, Mathematics and Computer Science)
J. Wen – Mentor (TU Delft - Algorithmics)
T.E.P.M.F. Abeel – Mentor (TU Delft - Pattern Recognition and Bioinformatics)
Azqa Nadeem – Graduation committee member (TU Delft - Cyber Security)
More Info
expand_more
Other than for strictly personal use, it is not permitted to download, forward or distribute the text or part of it, without the consent of the author(s) and/or copyright holder(s), unless the work is under an open content license such as Creative Commons.
Abstract
To reduce food waste, the strawberry harvesting process should be optimized. In the modern era, computer vision can provide huge amounts of help. This paper focuses on optimizing pre-trained convolutional neural networks (CNN) to determine the maturity level of strawberries on a 1-10 scale. Here, 1 means unripe and 10 means overripe. Maturity level 8 is marketable. Experiments are done with VGG19, Resnet50, InceptionV2, Alexnet, and EfficientNetB2 as classifiers on segments using ADAM and SGD as optimizers and cross-entropy as loss function. The same CNN's are applied as a backbone for FasterRCNN to see how they would behave within an object detection architecture. The biggest challenge during this research was the low amount of training data. The research showed that using convolutional neural networks as a maturity level predictor is possible, but a well made training set with an equal spread for each maturity level is necessary to possibly achieve high accuracy.