Face Mask Recognition based on YOLOV3

Abstract

Wearing masks is considered one of the effective ways to prevent epidemics and protect physical health. In densely populated public places, detecting whether the mask is correctly worn has important application value, which can help managers reduce the potential risk of disease transmission. Previous works mainly rely on manual features, which are limited to fully expressing complex scene information. Thanks to the rapid development of artificial intelligence technology, automatic mask detection methods based on deep learning have achieved breakthroughs in speed and accuracy, while mask detection in complex scenes is still an open issue. To alleviate this issue, this paper proposes a mask detection method based on YOLOV3. Specifically, this study introduces regularization methods such as Drop Block and Label Smoothing, as well as data augmentation strategies such as Gridmask and Mosaic. This study further explores the recognition performance for refined scenarios including no mask, standard mask-wearing and incorrect mask-wearing. Extensive experimental results show the effectiveness of our method. By introducing the improved YOLOV5 algorithm to realize the mask detection function, the data structure is optimized through pruning, the data volume is reduced, the calculation speed is accelerated, and the mask detection is well realized.