Early bruise detection, classification and prediction in strawberry using Vis-NIR hyperspectral imaging

Abstract

The most frequent kind of damage to strawberries is bruising. However, most of the bruises are so barely perceptible at an early stage on the surface, that detection of them with the human eye is quite challenging. This study proposes a method for accurately detecting and classifying the damage using reflectance imaging spectroscopy. In order to carry out the study, an experiment was devised to artificially induce bruises and a dataset was generated at different bruise intervals. A model for detecting and classifying bruises at their latent stage was developed using machine learning classifiers, including support vector machines (SVM), k-nearest neighbors (KNN), linear discriminant analysis (LDA), random forest (RF), and decision tree (DT), to investigate the changes over time after bruise occurrence on the detection performance. Regression models for the prediction of bruising time were developed using partial least square regression (PLSR), RF, gradient boosting (GB), support vector regression (SVR), and DT. Among the compared models, both SVM and LDA could achieve 99.99 % classification accuracy. RF was regarded as being the most advisable for detection and prediction jobs due to its high performance. It achieved MSE of 0.052 and R2 of 0.989 for prediction.