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

被引:165
作者
Bansal, Monika [1 ]
Kumar, Munish [2 ]
Sachdeva, Monika [3 ]
Mittal, Ajay [4 ]
机构
[1] SSD Women Inst Technol, Bathinda, Punjab, India
[2] Maharaja Ranjit Singh Punjab Tech Univ, Dept Computat Sci, Bathinda, Punjab, India
[3] IKG Punjab Tech Univ, Dept Comp Sci & Engn, Mohali Campus 1, Mohali, India
[4] Panjab Univ, Univ Inst Engn & Technol, Dept Comp Sci & Engn, Chandigarh, India
关键词
ORB; SIFT; SURF; K-Means; LPP; PCA; FEATURES;
D O I
10.1007/s12652-021-03488-z
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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 Naive 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.
引用
收藏
页码:3609 / 3620
页数:12
相关论文
共 24 条
[1]  
[Anonymous], P 3 INT C LEARNING R
[2]  
[Anonymous], 2008, COMPUT VIS IMAGE UND, DOI DOI 10.1016/j.cviu.2007.09.014
[3]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[4]  
Garg A, 2020, AAAI CONF ARTIF INTE, V34, P13789
[5]  
Harris C., 1988, P ALV VIS C, V15, P10, DOI [10.5244/C.2.23, DOI 10.5244/C.2.23]
[6]   Transfer learning for decision support in Covid-19 detection from a few images in big data [J].
Karthikeyan, Divydharshini ;
Varde, Aparna S. ;
Wang, Weitian .
2020 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2020, :4873-4881
[7]  
Kataoka H, 2015, FEATURE EVALUATION D
[8]  
Kleinberg EM, 1996, ANN STAT, V24, P2319
[9]   Ensembling Classical Machine Learning and Deep Learning Approaches for Morbidity Identification From Clinical Notes [J].
Kumar, Vivek ;
Recupero, Diego Reforgiato ;
Riboni, Daniele ;
Helaoui, Rim .
IEEE ACCESS, 2021, 9 (09) :7107-7126
[10]   Deep Learning Based Automatic Multiclass Wild Pest Monitoring Approach Using Hybrid Global and Local Activated Features [J].
Liu, Liu ;
Xie, Chengjun ;
Wang, Rujing ;
Yang, Po ;
Sudirman, Sud ;
Zhang, Jie ;
Li, Rui ;
Wang, Fangyuan .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2021, 17 (11) :7589-7598