Enhancing COVID-19 Safety: Exploring YOLOv8 Object Detection for Accurate Face Mask Classification
Keywords:
Face mask detection, Deep Learning, YOLO v8Abstract
These The COVID-19 pandemic has emphasized the importance of wearing face masks as an effective measure to reduce the spreading of the virus. With the increasing demand for automated systems capable of detecting and classifying face mask wearing conditions, deep learning models have emerged as a powerful tool in this domain. In this research paper, we investigate the performance of the YOLOv8 (You Only Look Once) object detection algorithm for the classification of face mask wearing conditions. YOLOv8 is a state-of-the-art deep learning model known for its real-time object detection capabilities. The model is trained with Face Mask Detector(FMD) dataset to provide ground truth labels for training and evaluation purposes. We fine-tune the YOLOv8 model using transfer learning techniques on this dataset, enabling it to classify face mask wearing conditions accurately. The experiments performed demonstrate that the YOLOv8 model achieves excellent performance in face mask wearing condition classification. We evaluate the model on various metrics, including precision, recall, mAP, to assess its accuracy, sensitivity, and overall performance. The results show that the model successfully distinguishes between individuals wearing face masks, not wearing face masks, or wearing face masks incorrectly, with high precision and recall rates.The YOLOv5 model was also trained using the same dataset for comparative analysis.
Downloads
References
worldometers (accessed on 20th May,2023
H. Adusumalli, D. Kalyani, R. K. Sri, M. Pratapteja and P. V. R. D. P. Rao, ”Face Mask Detection Using OpenCV,” 2021 Third International Conference on Intelligent Communication Technologies and Virtual Mobile Networks (ICICV), 2021, pp. 1304-1309, doi: 10.1109/ICICV50876.2021.9388375.
Kaggle/datasets/spandanpatnaik09/face-mask-detector(accessed on March 2023)
Kumar, T. A., Rajmohan, R., Pavithra, M., Ajagbe, S. A., Hodhod, R., & Gaber, T. (2022). Automatic face mask detection system in public transportation in smart cities using IoT and deep learning. Electronics, 11(6), 904.
Ryumina, E., Ryumin, D., Ivanko, D., & Karpov, A. (2021). A NOVEL METHOD FOR PROTECTIVE FACE MASK DETECTION USING CONVOLUTIONAL NEURAL NETWORKS AND IMAGE HISTOGRAMS. International Archives of the Photogrammetry, Remote Sensing & Spatial Information Sciences.
Hu, Y., Xu, Y., Zhuang, H., Weng, Z., & Lin, Z. (2022). Machine Learning Techniques and Systems for Mask-Face Detection—Survey and a New OOD-Mask Approach. Applied Sciences, 12(18), 9171.
Lin, H.; Tse, R.; Tang, S.K.; Chen, Y.; Ke, W.; Pau, G. Near-Realtime Face Mask Wearing Recognition Based on Deep Learning. In Proceedings of the 2021 IEEE 18th Annual Consumer Communications & Networking Conference (CCNC), Las Vegas, NV, USA, 9–12 January 2021; pp. 1–7.
rusty, M.R.; Tripathi, V.; Dubey, A. A novel data augmentation approach for mask detection using deep transfer learning. Intell.-Based Med. 2021, 5, 100037
Han, Z.; Huang, H.; Fan, Q.; Li, Y.; Li, Y.; Chen, X. SMD-YOLO: An efficient and lightweight detection method for mask wearing status during the COVID-19 pandemic. Comput. Methods Programs Biomed. 2022, 221, 106888.
Ottakath, N.; Elharrouss, O.; Almaadeed, N.; Al-Maadeed, S.; Mohamed, A.; Khattab, T.; Abualsaud, K. ViDMASK dataset for face mask detection with social distance measurement. Displays 2022, 73, 102235.
Olukumoro, O.S.; Ajayi, F.A.; Adebayo, A.A.; Usman, A.A.B.; Johnson, F. HIC-DEEP: A Hierarchical Clustered Deep Learning Model for Face Mask Detection. Int. J. Res. Innov. Appl. Sci. 2022, 7, 22–28
roboflow/whats-new-in-yolov8/(accessed on 19th April,2023)
Redmon, J., Divvala, S., Girshick, R., & Farhadi, A. (2016). You only look once: Unified, real-time object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 779- 788).
Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C. Y., & Berg, A. C. (2016). Ssd: Single shot multibox detector. In Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, October 11–14, 2016, Proceedings, Part I 14 (pp. 21- 37). Springer International Publishing.
labelImg/1.4.0/ (opensource)
Jindal, N., Singh, H., & Rana, P. S. (2022). Face mask detection in COVID-19: a strategic review. Multimedia Tools and Applications, 81(28), 40013-40042.
Parab, C. U., Mwitta, C., Hayes, M., Schmidt, J. M., Riley, D., Fue, K., ... & Rains, G. C. (2022). Comparison of Single-Shot and TwoShot Deep Neural Network Models for Whitefly Detection in IoT Web Application. AgriEngineering, 4(2), 507-522.
Girshick, R. (2015). Fast r-cnn. In Proceedings of the IEEE international conference on computer vision (pp. 1440-1448)
Khalili, S., & Shakiba, A. (2022, February). A face detection method via ensemble of four versions of YOLOs. In 2022 International Conference on Machine Vision and Image Processing (MVIP) (pp. 1-4). IEEE.
Li, R., & Yang, J. (2018, May). Improved YOLOv2 object detection model. In 2018 6th international conference on multimedia computing and systems (ICMCS) (pp. 1-6). IEEE.
Masurekar, O., Jadhav, O., Kulkarni, P.,& Patil, S. (2020). Real time object detection using YOLOv3. International Research Journal of Engineering and Technology (IRJET), 7(03), 3764-3768.
https://medium.com/deelvin-machine-learning/the-evolution-of-theyolo-neural-networks-family-from-v1-to-v7-48dd98702a3d(accessed on 1st march,2023)
Jeong, H. J., Park, K. S., & Ha, Y. G. (2018, January). Image preprocessing for efficient training of YOLO deep learning networks. In 2018 IEEE International Conference on Big Data and Smart Computing (BigComp) (pp. 635-637). IEEE.
Bochkovskiy, A., Wang, C. Y., & Liao, H. Y. M. (2020). Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934.
https://pub.towardsai.net/yolov8-is-here-and-it-gets-better54b12b87e3b9(accessed on 6th May 2023)
Liu, Y., Lu, B., Peng, J., & Zhang, Z. (2020). Research on the use of YOLOv5 object detection algorithm in mask wearing recognition. World Scientific Research Journal, 6(11), 276-284.
Ahmad, M., Abbas, S., Fatima, A., Issa, G. F., Ghazal, T. M., & Khan, M. A. (2023). Deep Transfer Learning-Based Animal Face Identification Model Empowered with Vision-Based Hybrid Approach. Applied Sciences, 13(2), 1178.
Terven, J., & Cordova-Esparza, D. (2023). A Comprehensive Review of YOLO: From YOLOv1 to YOLOv8 and Beyond. arXiv preprint arXiv:2304.00501.
Hao, W., & Zhili, S. (2020, November). Improved mosaic: algorithms for more complex images. In Journal of Physics: Conference Series (Vol. 1684, No. 1, p. 012094). IOP Publishing.
B. Qin, D. Li, Identifying facemask-wearing condition using image super-resolution with classification network to prevent COVID-19, Sensors (Switzerland) (2020), doi:10.3390/s20185236
GJ. Chowdary, NS. Punn, SK. Sonbhadra, S. Agarwal, Face mask detection using transfer learning of inceptionV3, ArXiv (2020), doi:10.1007/978-3-030-66665-1-6.
P. Nagrath, R. Jain, A. Madan, R. Arora, P. Kataria, J. Hemanth, SSDMNV2: a real time DNN-based face mask detection system using single shot multibox detector and MobileNetV2, Sustain. Cities Soc. (2021), doi:10.1016/j.scs.2020.102692
M. Loey, G. Manogaran, MHN. Taha, NEM. Khalifa, Fighting against COVID-19: a novel deep learning model based on YOLO-v2 with ResNet-50 for medical face mask detection, Sustain. Cities Soc. (2021), doi:10.1016/j.scs.2020.102600.
S. Taneja, A. Nayyar, Vividha, P. Nagrath, in: Face Mask Detection Using Deep Learning During COVID-19, Springer, Singapore, 2021, pp. 39–51, doi:10.1007/978-981-16-0733-2-3.
HC. G, J. J, A. K, KM. Sagayam, CNN-based mask detection system using openCV and MobileNetV2, in: 2021 3rd Int. Conf. Signal Process. Commun., IEEE, 2021, pp. 115–119, doi:10.1109/ICSPC51351.2021.9451688.
RK. Kodali, R. Dhanekula, Face mask detection using deep learning, 2021 Int. Conf. Comput. Commun. Informatics, ICCCI 2021, Institute of Electrical and Electronics Engineers Inc, 2021, doi:10.1109/ICCCI50826.2021.9402670.
SK. Addagarla, G. Kalyan Chakravarthi, P. Anitha, Real time multiscale facial mask detection and classification using deep transfer learning techniques, Int. J. Adv. Trends Computer. Sci. Eng. (2020), doi:10.30534/ijatcse/2020/33942020.
He, K.; Zhang, X.; Ren, S.; Sun, J. Deep Residual Learning for Image Recognition. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 27–30 June 2016
Kumar, A., Kalia, A., Verma, K., Sharma, A., Kaushal, M. (2021). Scaling up face masks detection with YOLO on a novel dataset. Optik, 239, 166744. doi:10.1016/j.ijleo.2021.166744
Degadwala, D. Vyas, U. Chakraborty, A. R. Dider and H. Biswas, ”Yolov4 Deep Learning Model for Medical Face Mask Detection,” 2021 International Conference on Artificial Intelligence and Smart Systems (ICAIS), 2021, pp. 209-213, doi: 10.1109/ICAIS50930.2021.9395857.
S. Abbasi, H. Abdi and A. Ahmadi, ”A Face-Mask Detection Approach based on YOLO Applied for a New Collected Dataset,” 2021 26th International Computer Conference, Computer Society of Iran (CSICC), 2021, pp. 1-6, doi: 10.1109/CSICC52343.2021.9420599.
Downloads
Published
How to Cite
Issue
Section
License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
All papers should be submitted electronically. All submitted manuscripts must be original work that is not under submission at another journal or under consideration for publication in another form, such as a monograph or chapter of a book. Authors of submitted papers are obligated not to submit their paper for publication elsewhere until an editorial decision is rendered on their submission. Further, authors of accepted papers are prohibited from publishing the results in other publications that appear before the paper is published in the Journal unless they receive approval for doing so from the Editor-In-Chief.
IJISAE open access articles are licensed under a Creative Commons Attribution-ShareAlike 4.0 International License. This license lets the audience to give appropriate credit, provide a link to the license, and indicate if changes were made and if they remix, transform, or build upon the material, they must distribute contributions under the same license as the original.
