Abstract

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.

Highlights

  • The COVID-19 pandemic became a major global problem in 2020

  • Dist is the type of distance, NN is the Number of neighbours, AVG is the average metric value, and AVG Perf is the sum of the average metric accuracy, precision, recall, and F1 score divided by 4

  • It can be seen that there is no significant difference in performance with the parameters used in the k-nearest neighbors (KNN)

Read more

Summary

Introduction

The COVID-19 pandemic became a major global problem in 2020. The implementation of health protocols is still the most effective way to prevent the transmission, such as by wearing face masks. 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. The classification model can be implemented for automatic alert system to detect and warn people who are not wearing face masks

Objectives
Methods
Results
Discussion
Conclusion
Full Text
Published version (Free)

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call