1. Banana fruit classification using computer vision.
- Author
-
Makain, Felice Maynelia, Shah, Noraisyah Mohamed, and Mokhtar, Norrima
- Subjects
BANANAS ,COMPUTER vision ,FRUIT ,CONVOLUTIONAL neural networks ,DATA augmentation ,GRAYSCALE model - Abstract
Automatic banana fruit classification system was developed utlising pre-existing convolutional neural networks (CNN) models via transfer learning. The proposed model has the ability to classify five types of local banana fruits; Awak, Berangan, Cavendish, Lemak Manis, and Mas bananas. Dataset of 627 images was created to train our computer vision-based model. Data augmentation involving grayscale conversion, image zooming, horizontal and vertical flip, 90-degree rotation, and brightness adjustment is used to increase the total number of images to 6270. The augmented images are separated into three datasets, namely train, validation, and test datasets, with 4389 training images, 1254 validation images, and 627 test images respectively. Three CNN architectures including VGG16, InceptionV3, and ResNet152V2 were trained, and their performances evaluated. All three of the models are optimized by conducting several experiments to find the best parameters for learning rate, batch size, input size, and number of epochs. The best CNN architecture is found to be ResNet152V2 with a test accuracy of 94%, training accuracy of 94%, training loss of 14%, precision of 95%, recall of 94%, and an F1-score of 94%. The ResNet152V2 model is then integrated into our automated banana fruit classification system to predict the class of an input banana image. Overall, the automated banana fruit classification system performed well during the testing process by using new local banana fruit images. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF