Academic Journal of Computing & Information Science, 2022, 5(1); doi: 10.25236/AJCIS.2022.050113.
Ruisha Zhu, Linjun Zhao
School of Computer Science and Information Engineering, Hubei University, Wuhan, Hubei, 430062, China
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.
Yolov4-Tiny, Face Mask Detection Algorithm, Iou
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.
[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.