An Automated Face-mask Detection System using YOLOv5 for Preventing Spread of COVID-19
Keywords:Mask detection, Object detection, Image classification, YOLOv5, COVID-19, Pandemic
Object detection systems based on deep learning have been immensely successful in
complex object detection tasks images and have shown potential in a wide range of real-life applications
including the COVID-19 pandemic. One of the key challenges in containing and mitigating the infection
among the population is to ensure and enforce the proper use of face masks. The objective of this paper
is to detect the proper use of facial masks among the urban population in a megacity. In this study, we
trained and validated a new dataset to detect images such as ‘with mask’, ‘without mask’, and ‘mask
not in position’ using YOLOv5. The dataset is comprised of 6550 images with the three classes. The
dataset attained a commendable performance accuracy of 95% on mAP. This study can be implemented
for automated scanning for monitoring the proper use of face masks in different settings of public spaces.
Chiu, N. C., Chi, H., Tai, Y. L., Peng, C. C., Tseng, C. Y., Chen, C. C., Tan, B. F., & Lin, C. Y., “Impact of Wearing Masks, Hand Hygiene, and Social Distancing on Influenza, Enterovirus, and AllCause Pneumonia During the Corona irus Pandemic: Retrospective National Epidemiological Surveillance Study,” Journal of medical Internet research, 2020, 22(8), e21257. https://doi.org/10.2196/21257.
Kebede Y, Yitayih Y, Birhanu Z, Mekonen S, Ambelu A., “Knowledge, perceptions and preventive practices towards COVID-19 early in the outbreak among Jimma university medical center visitors, Southwest Ethiopia,” PLoS ONE, 2020, 15(5): e0233744. https://doi.org/10.1371/journal.pone.0233744.
Martinelli, L., Kopilaš, V., Vidmar, M., Heavin, C., Machado, H., Todorovic, Z., Buzas, N., Pot, M., Prainsack, B., & Gajovi ´ c, S., “Face Masks During the COVID-19 Pandemic: A Simple Protection Tool With Many Meanings,” Frontiers in public health, 2021, 8, 606635. https://doi.org/10.3389/fpubh.2020.606635.
Kushwaha, S., Bahl, S., Bagha, A. K., Parmar, K. S., Javaid, M., Haleem, A., & Singh, R. P., “Significant applications of machine learning for COVID-19 pandemic,” Journal of Industrial Integration and Management, 2020, 5(04), 453-479.
Lalmuanawma, S., Hussain, J., & Chhakchhuak, L., “Applications of machine learning and artificial intelligence for Covid-19 (SARSCoV-2) pandemic: A review,” Chaos, Solitons & Fractals, 2020, 139, 110059.
Ivašic-Kos, M., Krišto, M., & Pobar, M., “Human detection in thermal imaging using YOLO,” In Proceedings of the 2019 5th International Conference on Computer and Technology Applications, 2019, pp. 20-24.
Z. Cai and N. Vasconcelos, “Cascade R-CNN: High Quality Object Detection and Instance Segmentation,” IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021, vol. 43, no. 5, pp. 1483-1498, 1 May 2021, doi: 10.1109/TPAMI.2019.2956516.
K. Bhambani, T. Jain and K. A. Sultanpure, “Real-time Face Mask and Social Distancing Violation Detection System using YOLO,” 2020 IEEE Bangalore Humanitarian Technology Conference (BHTC), 2020, pp. 1-6, doi: 10.1109/B-HTC50970.2020.9297902.
P. Hurtik, V. Molek and P. Vlasanek, “YOLO-ASC: You Only Look Once And See Contours,” 2020 International Joint Conference on Neural Networks (IJCNN), 2020, pp. 1-7, doi: 10.1109/IJCNN48605.2020.9207223.
Vajgl, M.; Hurtik, P.; Nejezchleba, T., “Dist-YOLO: Fast Object Detection with Distance Estimation,” Appl. Sci., 2022, 12, 1354. https://doi.org/10.3390/app12031354.
Hardan, F., & R. J. Almusawi, A., “Developing an Automated Vision System for Maintaing Social Distancing to Cure the Pandemic,” Al-Khwarizmi Engineering Journal (Alkej), 2022, 18(1), 38–50. https://doi.org/10.22153/kej.2022.03.002.
Sengupta, K., Srivastava, P.R., “HRNET: AI-on-Edge for Mask Detection and Social Distancing Calculation,” SN COMPUT. SCI., 2022, 3, 157 (2022). https://doi.org/10.1007/s42979-022-01023-1
Karacı A., “VGGCOV19-NET: automatic detection of COVID-19 cases from X-ray images using modified VGG19 CNN architecture and YOLO algorithm,” Neural Comput Appl., 2022, Jan 24:1-22. doi: 10.1007/s00521-022-06918-x. Epub ahead of print. PMID: 35095212; PMCID: PMC8785935.
Sozzi, M.; Cantalamessa, S.; Cogato, A.; Kayad, in A.; White Marinello, F., “Automatic Bunch Detection Grape Varieties Using YOLOv3, YOLOv4, and YOLOv5 Deep Learning Algorithms,” Agronomy, 2022, 12, 319. https://doi.org/10.3390/agronomy12020319.
da Silva, J.R.; de Almeida, G.M.; Cuadros, M.A.d.S.L.; Campos, H.L.M.; Nunes, R.B.; Simão, J.; Muniz, P.R., “Recognition of Human Face Regions under Adverse Conditions – Face Masks and Glasses – In Thermographic Sanitary Barriers through Learning Transfer from an Object Detector,” Machines, 2022, 10, 43. https://doi.org/10.3390/machines10010043.
Imam Al Amin and F. Arby, “Implementation of YOLO-v5 for a Real Time Social Distancing Detection,” JAIC, 2022, vol. 6, no. 1, pp. 01-06.
Kim, J.-H.; Kim, N.; Park, Y.W.; Won, C.S., “Object Detection and Classification Based on YOLO-V5 with Improved Maritime Dataset,” J. Mar. Sci. Eng, 2022, 10, 377. https://doi.org/10.3390/jmse10030377.
Degadwala, S., Vyas, D., Chakraborty, U., Dider, A. R., & Biswas, H., “Yolo-v4 deep learning model for medical face mask detection,” 2021 International Conference on Artificial Intelligence and Smart Systems (ICAIS), 2021, (pp. 209-213). IEEE.
K. V. Sathyamurthy, A. R. Shri Rajmohan, A. Ram Tejaswar, K. V and G. Manimala, “Realtime Face Mask Detection Using TINYYOLO V4,” 2021 4th International Conference on Computing and Communications Technologies (ICCCT), 2021, pp. 169-174, doi: 10.1109/ICCCT53315.2021.9711838.
Jian, W., & Lang, L., “Face mask detection based on Transfer learning and PP-YOLO,” In 2021 IEEE 2nd International Conference on Big Data, Artificial Intelligence and Internet of Things Engineering (ICBAIE), 2021, (pp. 106-109). IEEE.
S. Susanto, F. A. Putra, R. Analia and I. K. L. N. Suciningtyas, “The Face Mask Detection For Preventing the Spread of COVID-19 at Politeknik Negeri Batam,” 2020 3rd International Conference on Applied Engineering (ICAE), 2020, pp. 1-5, doi: 10.1109/ICAE50557.2020.9350556.
M. R. Bhuiyan, S. A. Khushbu and M. S. Islam, “A Deep Learning Based Assistive System to Classify COVID-19 Face Mask for Human Safety with YOLOv3,” 2020 11th International Conference on Computing, Communication and Networking Technologies (ICCCNT), 2020, pp. 1-5, doi: 10.1109/ICCCNT49239.2020.9225384.
Liu, R., & Ren, Z., “Application of Yolo on Mask Detection Task,” In 2021 IEEE 13th International Conference on Computer Research and Development (ICCRD), 2021, (pp. 130-136). IEEE.
Li, Y., Yan, J., & Hu, B., “Mask Detection Based On EfficientYOLO” In 2021 40th Chinese Control Conference (CCC), 2021, (pp. 4056-4061). IEEE.
Singh, S., Ahuja, U., Kumar, M., Kumar, K., & Sachdeva, M., “Face mask detection using YOLOv3 and faster R-CNN models: COVID-19 environment,” Multimedia Tools and Applications, 2021, Spring: 80(13), 19753-19768
Loey, M., Manogaran, G., Taha, M. H. N., & Khalifa, N. E. M., “Fighting against COVID-19: A novel deep learning model based on YOLO-v2 with ResNet-50 for medical face mask detection,” Sustainable cities and society, 2021, 65, 102600.
Kumar A, Kalia A, Sharma A, Kaushal M., “A hybrid tiny YOLO v4-SPP module based improved face mask detection vision system,” J Ambient Intell Humaniz Comput, 2021, 20:1-14. doi: 10.1007/s12652-021-03541-x.
Yu, J., & Zhang, W., “Face mask wearing detection algorithm based on improved YOLO-v4,” Sensors, 2021, 21(9), 3263.
Balaji, S., Balamurugan, B., Kumar, T. A., Rajmohan, R., & Kumar, P. P., “A brief Survey on AI Based Face Mask Detection System for Public Places,” Irish Interdisciplinary Journal of Science & Research (IIJSR), 2021.
Prasetia, D. D., Yuswanto, A., & Wibowo, B., “Design of machine learning detection mask using yolo and darknet on NVIDIA Jetson Nano,” TEKNOKOM, 2022, 5(1), 88-95.
Z. Qin, Z. Guo and Y. Lin, “An Implementation of Face Mask Detection System Based on YOLOv4 Architecture,” 14th International Conference on Computer Research and Development (ICCRD), 2022, pp. 207-213, doi: 10.1109/ICCRD54409.2022.9730470.
J. Han and M. Kamber, “Data Mining: Concepts and Techniques,” 2nd ed., Morgan Kaufmann, 2006.
Mohammed J. Zaki, Wagner Meira JR, “Data Mining and Analysis,” Fundamental Concepts and Algorithms: Campridge University Press, 2014.
Masum, M. Islam; Jishan, Md Asifuzzaman; Mahmud, Khan Raqib; Azad, Abul Kalam Al, “Covid Face-Mask Monitoring Dataset,” Harvard Dataverse, https://doi.org/10.7910/DVN/BKPMHT.
How to Cite
LicenseInternational Journal of Computing is an open access journal. Authors who publish with this journal agree to the following terms:
• Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
• Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
• Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work.