Mask Wearing Detection Algorithm Based on Improved Tiny YOLOv3

摘要

The new coronavirus spreads widely through droplets, aerosols and other carriers. Wearing a mask can effectively reduce the probability of being infected by the virus. Therefore, it is necessary to monitor whether people wear masks in public to prevent the virus from spreading further. However, there is no mature general mask wearing detection algorithm. Based on tiny YOLOv3 algorithm, this paper realizes the detection of face with mask and face without mask, and proposes an improvement to the algorithm. First, the loss function of the bounding box regression is optimized, and the original loss function is optimized as the Generalized Intersection over Union (GIoU) loss. Second, the network structure is improved, the residual unit is introduced into the backbone to increase the depth of the network and the detection of two scales is expanded to three. Finally, the size of anchor boxes is clustered based on k-means algorithm. The experimental results on the constructed dataset show that, compared with the tiny YOLOv3 algorithm, the algorithm proposed in this paper improves the detection accuracy while maintaining high-speed inference ability.