Prediction of Total Soluble Solids and pH of Strawberry Fruits Using RGB, HSV and HSL Colour Spaces and Machine Learning Models

Abstract

Determination of internal qualities such as total soluble solids (TSS) and pH is a paramount concern in strawberry cultivation. Therefore, the main objective of the current study was to develop a non-destructive approach with machine learning algorithms for predicting TSS and pH of strawberries. Six hundred samples (100 samples in each ripening stage) in six ripening stages were collected randomly for measuring the biometrical characteristics, i.e., length, diameters, weight and TSS and pH values. An image of each strawberry fruit was captured for colour feature extraction using an image processing technique. Channels of each colour space (RGB, HSV and HSL) were used as input variables for developing multiple linear regression (MLR) and support vector machine regression (SVM-R) models. The result of the study indicated that SVM-R model with HSV colour space performed slightly better than MLR model for TSS and pH prediction. The HSV based SVM-R model could explain a maximum of 84.1% and 79.2% for TSS and 78.8% and 72.6% for pH of the variations in measured and predicted data in training and testing stages, respectively. Further experiments need to be conducted with different strawberry cultivars for the prediction of more internal qualities along with the improvement of model performance.