Tropical fruits classification using an alexnet-type convolutional neural network and image augmentation

Alberto Patino-Saucedo, Horacio Rostro-Gonzalez, Jorg Conradt

Producció científica: Capítol de llibreContribució a congrés/conferènciaAvaluat per experts

14 Cites (Scopus)

Resum

AlexNet is a Convolutional Neural Network (CNN) and reference in the field of Machine Learning for Deep Learning. It has been successfully applied to image classification, especially in large sets such as ImageNet. Here, we have successfully applied a smaller version of the AlexNet CNN to classify tropical fruits from the Supermarket Produce dataset. This database contains 2633 images of fruits divided into 15 categories with high variability and complexity, i.e. shadows, pose, occlusion, reflection (fruits inside a bag), etc. Since few training samples are required for fruit classification and to prevent overfitting, the modified AlexNet CNN has fewer feature maps and fully connected neurons than the original one, and data augmentation of the training set is used. Numerical results show a top-1 classification accuracy of 99.56 %, and a top-2 accuracy of 100 % for the 15 classes, which outperforms previous works on the same dataset.

Idioma originalAnglès
Títol de la publicacióNeural Information Processing - 25th International Conference, ICONIP 2018, Proceedings
EditorsSeiichi Ozawa, Andrew Chi Sing Leung, Long Cheng
EditorSpringer Verlag
Pàgines371-379
Nombre de pàgines9
ISBN (imprès)9783030042110
DOIs
Estat de la publicacióPublicada - 2018
Publicat externament
Esdeveniment25th International Conference on Neural Information Processing, ICONIP 2018 - Siem Reap, Cambodia
Durada: 13 de des. 201816 de des. 2018

Sèrie de publicacions

NomLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volum11304 LNCS
ISSN (imprès)0302-9743
ISSN (electrònic)1611-3349

Conferència

Conferència25th International Conference on Neural Information Processing, ICONIP 2018
País/TerritoriCambodia
CiutatSiem Reap
Període13/12/1816/12/18

Fingerprint

Navegar pels temes de recerca de 'Tropical fruits classification using an alexnet-type convolutional neural network and image augmentation'. Junts formen un fingerprint únic.

Com citar-ho