Face Mask Wearing Detection Algorithm Based on Improved YOLO-v4

Sensors (Basel). 2021 May 8;21(9):3263. doi: 10.3390/s21093263.

Abstract

To solve the problems of low accuracy, low real-time performance, poor robustness and others caused by the complex environment, this paper proposes a face mask recognition and standard wear detection algorithm based on the improved YOLO-v4. Firstly, an improved CSPDarkNet53 is introduced into the trunk feature extraction network, which reduces the computing cost of the network and improves the learning ability of the model. Secondly, the adaptive image scaling algorithm can reduce computation and redundancy effectively. Thirdly, the improved PANet structure is introduced so that the network has more semantic information in the feature layer. At last, a face mask detection data set is made according to the standard wearing of masks. Based on the object detection algorithm of deep learning, a variety of evaluation indexes are compared to evaluate the effectiveness of the model. The results of the comparations show that the mAP of face mask recognition can reach 98.3% and the frame rate is high at 54.57 FPS, which are more accurate compared with the exiting algorithm.

Keywords: CSPDarknNet53; PANet; YOLO-v4; adaptive image scaling; face mask recognition.

MeSH terms

  • Algorithms
  • COVID-19*
  • Facial Recognition*
  • Humans
  • Masks
  • Recognition, Psychology

Grants and funding