Comparative Analysis of Image Classification Algorithms for Face Mask Detection
Downloads
Background: The COVID-19 pandemic remains a problem in 2021. Health protocols are needed to prevent the spread, including wearing a face mask. Enforcing people to wear face masks is tiring. AI can be used to classify images for face mask detection. There are a lot of image classification algorithm for face mask detection, but there are still no studies that compare their performance.
Objective: This study aims to compare the classification algorithms of classical machine learning. They are k-nearest neighbors (KNN), support vector machine (SVM), and a widely used deep learning algorithm for image classification which is convolutional neural network (CNN) for face masks detection.
Methods: This study uses 5 and 3 cross-validation for assessing the performance of KNN, SVM, and CNN in face mask detection.
Results: CNN has the best average performance with the accuracy of 0.9683 and average execution time of 2,507.802 seconds for classifying 3,725 faces with mask and 3,828 faces without mask images.
Conclusion: For a large amount of image data, KNN and SVM can be used as temporary algorithms in face mask detection due to their faster execution times. At the same time, CNN can be trained to form a classification model. In this case, it is advisable to use CNN for classification because it has better performance than KNN and SVM. In the future, the classification model can be implemented for automatic alert system to detect and warn people who are not wearing face masks.
S. Zhang, X. Li, M. Zong, X. Zhu, and D. Cheng, “Learning k for kNN Classification,” ACM Trans. Intell. Syst. Technol., vol. 8, no. 3, 2017, doi: 10.1145/2990508.
X. Wu et al., Top 10 algorithms in data mining, vol. 14, no. 1. 2008.
N. O’Mahony et al., “Deep Learning vs. Traditional Computer Vision,” Adv. Intell. Syst. Comput., vol. 943, no. Cv, pp. 128–144, 2020, doi: 10.1007/978-3-030-17795-9_10.
K. O’Shea and R. Nash, “An Introduction to Convolutional Neural Networks,” pp. 1–11, 2015, [Online]. Available: http://arxiv.org/abs/1511.08458.
A. A. M. Al-Saffar, H. Tao, and M. A. Talab, “Review of deep convolution neural network in image classification,” Proceeding - 2017 Int. Conf. Radar, Antenna, Microwave, Electron. Telecommun. ICRAMET 2017, vol. 2018-Janua, pp. 26–31, 2017, doi: 10.1109/ICRAMET.2017.8253139.
M. Pak and S. Kim, “A review of deep learning in image recognition,” Proc. 2017 4th Int. Conf. Comput. Appl. Inf. Process. Technol. CAIPT 2017, vol. 2018-Janua, pp. 1–3, 2018, doi: 10.1109/CAIPT.2017.8320684.
E. Setiawan and A. Muttaqin, “Implementation of K-Nearest Neighbors face recognition on low-power processor,” Telkomnika (Telecommunication Comput. Electron. Control., vol. 13, no. 3, pp. 949–954, 2015, doi: 10.12928/telkomnika.v13i3.713.
P. M. Shah, “Face detection from images using support vector machine,” 2012.
M. T. Ghazal and K. Abdullah, “Face recognition based on curvelets, invariant moments features and SVM,” Telkomnika (Telecommunication Comput. Electron. Control., vol. 18, no. 2, pp. 733–739, 2020, doi: 10.12928/TELKOMNIKA.v18i2.14106.
A. R. Syafeeza, M. Khalil-Hani, S. S. Liew, and R. Bakhteri, “Convolutional neural network for face recognition with pose and illumination variation,” Int. J. Eng. Technol., vol. 6, no. 1, pp. 44–57, 2014.
C. Jagadeeswari and M. U. Theja, “Performance Evaluation of Intelligent Face Mask Detection System with various Deep Learning Classifiers Keywords :,” Int. J. Adv. Sci. Technol., vol. 29, no. 11, pp. 3074–3082, 2020.
W. Han, Z. Huang, A. Kuerban, M. Yan, and H. Fu, “A Mask Detection Method for Shoppers under the Threat of COVID-19 Coronavirus,” in Proceedings - 2020 International Conference on Computer Vision, Image and Deep Learning, CVIDL 2020, Jul. 2020, pp. 442–447, doi: 10.1109/CVIDL51233.2020.00-54.
V. Vinitha and V. Velantina, “Covid-19 Facemask Detection With Deep Learning and Computer Vision,” Int. Res. J. Eng. Technol., vol. 7, no. 8, pp. 3127–3132, 2020.
P. Nagrath, R. Jain, A. Madan, R. Arora, P. Kataria, and J. Hemanth, “SSDMNV2: A real time DNN-based face mask detection system using single shot multibox detector and MobileNetV2,” Sustain. Cities Soc., vol. 66, no. December 2020, 2021, doi: 10.1016/j.scs.2020.102692.
S. Ge, J. Li, Q. Ye, and Z. Luo, “Detecting masked faces in the wild with LLE-CNNs,” Proc. - 30th IEEE Conf. Comput. Vis. Pattern Recognition, CVPR 2017, vol. 2017-Janua, pp. 426–434, 2017, doi: 10.1109/CVPR.2017.53.
M. Grassi and M. Faundez-Zanuy, “Face recognition with facial mask application and neural networks,” Lect. Notes Comput. Sci. (including Subser. Lect. Notes Artif. Intell. Lect. Notes Bioinformatics), vol. 4507 LNCS, pp. 709–716, 2007, doi: 10.1007/978-3-540-73007-1_85.
A. Shustanov and P. Yakimov, “CNN Design for Real-Time Traffic Sign Recognition,” Procedia Eng., vol. 201, pp. 718–725, 2017, doi: 10.1016/j.proeng.2017.09.594.
Larxel, “Face Mask Detection | Kaggle,” 2020. https://www.kaggle.com/omkargurav/face-mask-dataset (accessed Feb. 01, 2021).
F. Chollet and & O., “Keras: the Python deep learning API,” Keras: the Python deep learning API, 2020. https://keras.io/ (accessed Dec. 18, 2020).
T. F. Gonzalez, “Handbook of approximation algorithms and metaheuristics,” Handb. Approx. Algorithms Metaheuristics, pp. 1–1432, 2007, doi: 10.1201/9781420010749.
C. Szegedy et al., “Going deeper with convolutions,” Proc. IEEE Comput. Soc. Conf. Comput. Vis. Pattern Recognit., vol. 07-12-June, pp. 1–9, 2015, doi: 10.1109/CVPR.2015.7298594.
K. Simonyan and A. Zisserman, “Very deep convolutional networks for large-scale image recognition,” 3rd Int. Conf. Learn. Represent. ICLR 2015 - Conf. Track Proc., pp. 1–14, 2015.
M. D. Zeiler and R. Fergus, “Visualizing and understanding convolutional networks,” Lect. Notes Comput. Sci. (including Subser. Lect. Notes Artif. Intell. Lect. Notes Bioinformatics), vol. 8689 LNCS, no. PART 1, pp. 818–833, 2014, doi: 10.1007/978-3-319-10590-1_53.
K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” Proc. IEEE Comput. Soc. Conf. Comput. Vis. Pattern Recognit., vol. 2016-Decem, pp. 770–778, 2016, doi: 10.1109/CVPR.2016.90.
Authors who publish with this journal agree to the following terms:
All accepted papers will be published under a Creative Commons Attribution 4.0 International (CC BY 4.0) License. Authors retain copyright and grant the journal right of first publication. CC-BY Licenced means lets others to Share (copy and redistribute the material in any medium or format) and Adapt (remix, transform, and build upon the material for any purpose, even commercially).
