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...
Autores principales: | , , , |
---|---|
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 |