Colombian fruit and vegetables recognition using convolutional neural networks and transfer learning

O. I. Alvarez-Canchila, D. E. Arroyo-Pérez, A. Patiňo-Saucedo, H. Rostro González, A. Patĩo-Vanegas

Producció científica: Article en revista indexadaArticle de conferènciaAvaluat per experts

4 Cites (Scopus)

Resum

Automatic image recognition is a convenient option for labeling and categorizing fruits and vegetables in supermarkets. This paper proposes the design and implementation of an automatic classification system for Colombian fruits, by training a convolutional neural network. A database was created to train and test the system, which consisted of 4980 images, labeled in 22 classes, each corresponding to pictures of the same kind of fruit, trying to reproduce the variability of a real case scenario with occlusions, different positions, rotations, lightings, colors, etc., and the use of bags. On-training data augmentation was used to further increase the robustness of the model. Additionally, transfer learning was implemented by taking the parameters of a pretrained model used for fruit classification as the new initial parameters of the proposed convolutional network, achieving an increase of the classification accuracy compared with the same model when trained with random initial weights. The final classification accuracy of the network was 98.12% which matches the scores achieved on previous works that performed fruit classification on less challenging datasets. Considering top-3 classification we report an accuracy of 99.95%.

Idioma originalAnglès
Número d’article012020
RevistaJournal of Physics: Conference Series
Volum1547
Número1
DOIs
Estat de la publicacióPublicada - 18 de juny 2020
Publicat externament
Esdeveniment16th National Meeting on Optics, ENO 2019 and 7th Andean and Caribbean Conference on Optics and Its Applications, CANCOA 2019 - Monteria, Colombia
Durada: 26 de nov. 201930 de nov. 2019

Fingerprint

Navegar pels temes de recerca de 'Colombian fruit and vegetables recognition using convolutional neural networks and transfer learning'. Junts formen un fingerprint únic.

Com citar-ho