Title :
Automatic fruit classification using random forest algorithm
Author :
Zawbaa, Hossam M. ; Hazman, Maryam ; Abbass, Mona ; Hassanien, Aboul Ella
Author_Institution :
Fac. of Math. & Comput. Sci., Babes-Bolyai Univ., Cluj-Napoca, Romania
Abstract :
The aim of this paper is to develop an effective classification approach based on Random Forest (RF) algorithm. Three fruits; i.e., apples, Strawberry, and oranges were analysed and several features were extracted based on the fruits´ shape, colour characteristics as well as Scale Invariant Feature Transform (SIFT). A preprocessing stages using image processing to prepare the fruit images dataset to reduce their color index is presented. The fruit image features is then extracted. Finally, the fruit classification process is adopted using random forests (RF), which is a recently developed machine learning algorithm. A regular digital camera was used to acquire the images, and all manipulations were performed in a MATLAB environment. Experiments were tested and evaluated using a series of experiments with 178 fruit images. It shows that Random Forest (RF) based algorithm provides better accuracy compared to the other well know machine learning techniques such as K-Nearest Neighborhood (K-NN) and Support Vector Machine (SVM) algorithms. Moreover, the system is capable of automatically recognize the fruit name with a high degree of accuracy.
Keywords :
agricultural products; cameras; feature extraction; image classification; image colour analysis; learning (artificial intelligence); shape recognition; transforms; MATLAB environment; RF algorithm; SIFT; apples; automatic fruit classification; classification approach; color index; feature extraction; fruit colour characteristics; fruit images dataset; fruit shape; image processing; images acquisition; machine learning algorithm; oranges; random forest algorithm; regular digital camera; scale invariant feature transform; strawberry; Accuracy; Feature extraction; Image color analysis; Radio frequency; Shape; Support vector machines; Training; Features extraction; Fruit classification; Image classification; Random Forest (RF); Scale Invariant Feature Transform (SIFT);
Conference_Titel :
Hybrid Intelligent Systems (HIS), 2014 14th International Conference on
Print_ISBN :
978-1-4799-7632-4
DOI :
10.1109/HIS.2014.7086191