Non-destructive Weight Prediction Model of Spherical Fruits and Vegetables using U-Net Image Segmentation and Machine Learning Methods

Autor: Halil Kayra, Savaş Koç
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: Journal of Agricultural Sciences, Vol 30, Iss 4, Pp 735-747 (2024)
Druh dokumentu: article
ISSN: 1300-7580
DOI: 10.15832/ankutbd.1434767
Popis: Artificial intelligence has become increasingly prominent in agriculture and other fields. Prediction of body weight in animals and plants has been done by humans using many different methods and observations from the past to the present. Although there has been extensive research on predicting the live body weight of animals, weight prediction of vegetables and fruits is not widely. As spherical or round-shaped fruits and vegetables are sold by weighing in the fields, markets and greengrocers, it is important to make weight predictions. Based on this, a model was developed to predict the weight of fruits and vegetables such as watermelons, melons, apples, oranges and tomatoes with the data obtained from their images. The fruit and vegetable weights were predicted by regression models using data obtained from images segmented by the U-Net architecture. Machine learning models such as Multi-Layer Perceptron (MLP), Random Forest (RF), Decision Trees (DT), Support Vector Machines (SVM), Linear and Stochastic Gradient Descent (SGD) regression models were used for weight predictions. The most effective regression models are the RF and DT models. For regression training, the best success rates were calculated as 0.9112 for watermelon, 0.9944 for apple, 0.9989 for tomato and 0.9996 for orange. In addition, the results were evaluated by comparing them to the studies of weight prediction. The weight prediction model will help to sell round-shaped fruits and vegetables in the fields, markets and gardens using the weight predictions from the images. It is also a guideline for studies that follow the growth of fruit and vegetables according to their weight.
Databáze: Directory of Open Access Journals