Welcome to Francis Academic Press

Academic Journal of Computing & Information Science, 2022, 5(1); doi: 10.25236/AJCIS.2022.050113.

Research and Application of Face Mask Detection Algorithm Based on YOLOV4-Tiny

Author(s)

Ruisha Zhu, Linjun Zhao

Corresponding Author:
Ruisha Zhu
Affiliation(s)

School of Computer Science and Information Engineering, Hubei University, Wuhan, Hubei, 430062, China

Abstract

This paper mainly studies the target detection of face masks. Based on yOLOV4-TINY algorithm, multi-scale feature fusion of FPN structure and feature enhancement network are used to optimize the algorithm structure. First of all, this paper uses the xml.etree.ElementTree (ET) module in python to read the xml file to obtain the key information such as < object > face parameters, < name > face mask wearing status, < bndbox > face bounding box diagonal two vertex coordinates and so on. According to the vertex coordinates (bndbox) in the xml file, the target box is marked on the original image, and the color state of the target box is distinguished according to the wearing state of < name > face mask. Finally, two pictures of "250.png" and "477.png" are selected to be displayed in the body of this text. Then, a face mask detection algorithm based on YOLOv4-Tiny is established. when training the model, the multi-task joint loss of mask detection task is optimized by combining CIOU loss function and label smoothing strategy, and the Mosaic data enhancement method and learning rate cosine annealing attenuation strategy are used to improve the convergence speed and robustness of the model. After many times of detection and verification, the experimental results show that the detection accuracy of the proposed algorithm model is 94.84% (with_mask), 96.04% (without_mask) and 94.86% (mask_weared_incorrect) respectively in the three categories of human face mask detection: with_mask, without_mask and mask_weared_incorrect. Its quantitative indicator < number of faces correctly classified > / < number of all faces included in the tag file >, mAP is as high as 95.25%, and the accuracy is greatly improved.

Keywords

Yolov4-Tiny, Face Mask Detection Algorithm, Iou

Cite This Paper

Ruisha Zhu, Linjun Zhao. Research and Application of Face Mask Detection Algorithm Based on YOLOV4-Tiny. Academic Journal of Computing & Information Science (2022), Vol. 5, Issue 1: 64-69. https://doi.org/10.25236/AJCIS.2022.050113.

References

[1] Ye Mao, Ma Jie, Wang Qian, Wu Lin. Lightweight mask wear detection algorithm based on multi-scale feature fusion [J]. Computer Engineering 2021 01000 (3428): 1-13.

[2] Wang C Y, Liao H Y M, Wu Y H, et al. CSPNet: A new backbone that can enhance learning capability of CNN [C] / / Proceedings of the 2020 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 14-19, 2020, Virtual, Online, United States: IEEE, 2020: 1571-1580.

[3] Wang Yihao, Ding Hongwei, Li Bo, Yang Zhijun, Yang Jundong. Mask wear detection algorithm based on improved YOLOv3 in complex scenes [J]. Computer Engineering, 2020 and 46 (11): 12-22.

[4] Ye Zixun, Zhang Hongying. Lightweight improvement of YOLOv4 mask detection algorithm [J]. Computer Engineering and applications, 2021 57 (17): 157-168.

[5] Rui Geng, Yixuan Ma, Wanhong Huang. An improved helmet detection method for YOLOv3 on an unbalanced dataset[C]. Proceedings of 2021 3rd International Conference on Advances in Computer Technology, Information Science and Communications (CTISC 2021). IEEE, 2021.