Cargando…

Transfer learning for image classification using VGG19: Caltech-101 image data set

Image classification is getting more attention in the area of computer vision. During the past few years, a lot of research has been done on image classification using classical machine learning and deep learning techniques. Presently, deep learning-based techniques have given stupendous results. Th...

Descripción completa

Detalles Bibliográficos
Autores principales: Bansal, Monika, Kumar, Munish, Sachdeva, Monika, Mittal, Ajay
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Springer Berlin Heidelberg 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8446720/
https://www.ncbi.nlm.nih.gov/pubmed/34548886
http://dx.doi.org/10.1007/s12652-021-03488-z
_version_ 1784568946487197696
author Bansal, Monika
Kumar, Munish
Sachdeva, Monika
Mittal, Ajay
author_facet Bansal, Monika
Kumar, Munish
Sachdeva, Monika
Mittal, Ajay
author_sort Bansal, Monika
collection PubMed
description Image classification is getting more attention in the area of computer vision. During the past few years, a lot of research has been done on image classification using classical machine learning and deep learning techniques. Presently, deep learning-based techniques have given stupendous results. The performance of a classification system depends on the quality of features extracted from an image. The better is the quality of extracted features, the more the accuracy will be. Although, numerous deep learning-based methods have shown enormous performance in image classification, still due to various challenges deep learning methods are not able to extract all the important information from the image. This results in a reduction in overall classification accuracy. The goal of the present research is to improve the image classification performance by combining the deep features extracted using popular deep convolutional neural network, VGG19, and various handcrafted feature extraction methods, i.e., SIFT, SURF, ORB, and Shi-Tomasi corner detector algorithm. Further, the extracted features from these methods are classified using various machine learning classification methods, i.e., Gaussian Naïve Bayes, Decision Tree, Random Forest, and eXtreme Gradient Boosting (XGBClassifier) classifier. The experiment is carried out on a benchmark dataset Caltech-101. The experimental results indicate that Random Forest using the combined features give 93.73% accuracy and outperforms other classifiers and methods proposed by other authors. The paper concludes that a single feature extractor whether shallow or deep is not enough to achieve satisfactory results. So, a combined approach using deep learning features and traditional handcrafted features is better for image classification.
format Online
Article
Text
id pubmed-8446720
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher Springer Berlin Heidelberg
record_format MEDLINE/PubMed
spelling pubmed-84467202021-09-17 Transfer learning for image classification using VGG19: Caltech-101 image data set Bansal, Monika Kumar, Munish Sachdeva, Monika Mittal, Ajay J Ambient Intell Humaniz Comput Original Research Image classification is getting more attention in the area of computer vision. During the past few years, a lot of research has been done on image classification using classical machine learning and deep learning techniques. Presently, deep learning-based techniques have given stupendous results. The performance of a classification system depends on the quality of features extracted from an image. The better is the quality of extracted features, the more the accuracy will be. Although, numerous deep learning-based methods have shown enormous performance in image classification, still due to various challenges deep learning methods are not able to extract all the important information from the image. This results in a reduction in overall classification accuracy. The goal of the present research is to improve the image classification performance by combining the deep features extracted using popular deep convolutional neural network, VGG19, and various handcrafted feature extraction methods, i.e., SIFT, SURF, ORB, and Shi-Tomasi corner detector algorithm. Further, the extracted features from these methods are classified using various machine learning classification methods, i.e., Gaussian Naïve Bayes, Decision Tree, Random Forest, and eXtreme Gradient Boosting (XGBClassifier) classifier. The experiment is carried out on a benchmark dataset Caltech-101. The experimental results indicate that Random Forest using the combined features give 93.73% accuracy and outperforms other classifiers and methods proposed by other authors. The paper concludes that a single feature extractor whether shallow or deep is not enough to achieve satisfactory results. So, a combined approach using deep learning features and traditional handcrafted features is better for image classification. Springer Berlin Heidelberg 2021-09-17 2023 /pmc/articles/PMC8446720/ /pubmed/34548886 http://dx.doi.org/10.1007/s12652-021-03488-z Text en © The Author(s), under exclusive licence to Springer-Verlag GmbH Germany, part of Springer Nature 2021 This article is made available via the PMC Open Access Subset for unrestricted research re-use and secondary analysis in any form or by any means with acknowledgement of the original source. These permissions are granted for the duration of the World Health Organization (WHO) declaration of COVID-19 as a global pandemic.
spellingShingle Original Research
Bansal, Monika
Kumar, Munish
Sachdeva, Monika
Mittal, Ajay
Transfer learning for image classification using VGG19: Caltech-101 image data set
title Transfer learning for image classification using VGG19: Caltech-101 image data set
title_full Transfer learning for image classification using VGG19: Caltech-101 image data set
title_fullStr Transfer learning for image classification using VGG19: Caltech-101 image data set
title_full_unstemmed Transfer learning for image classification using VGG19: Caltech-101 image data set
title_short Transfer learning for image classification using VGG19: Caltech-101 image data set
title_sort transfer learning for image classification using vgg19: caltech-101 image data set
topic Original Research
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8446720/
https://www.ncbi.nlm.nih.gov/pubmed/34548886
http://dx.doi.org/10.1007/s12652-021-03488-z
work_keys_str_mv AT bansalmonika transferlearningforimageclassificationusingvgg19caltech101imagedataset
AT kumarmunish transferlearningforimageclassificationusingvgg19caltech101imagedataset
AT sachdevamonika transferlearningforimageclassificationusingvgg19caltech101imagedataset
AT mittalajay transferlearningforimageclassificationusingvgg19caltech101imagedataset