Mask Wear Detection Algorithm Based on YOLOv5m

ABSTRACT

Wearing a mask during the COVID-19 epidemic can effectively prevent the spread of the virus. In view of the problems of small target size, crowd blocking each other and dense arrangement of targets in crowded places, a target detection algorithm based on the improved YOLOv5m model is proposed to achieve efficient detection of whether a mask is worn or not. This paper introduces four attention mechanisms in the feature extraction network based on the YOLOv5m model to suppress irrelevant information, enhance the information representation of the feature map, and improve the detection capability of the model for small-scale targets. The experimental results showed that the introduction of the SE module increased the mAP value of the original network by 9.3 percentage points, the most significant increase among the four attention mechanisms. And then a dual-scale feature fusion network is used in the Neck layer, giving different weights to the feature layers to convey more effective feature information. In the image pre-processing, the Mosaic method was used for data enhancement, and the CIoU loss function was used for coordinate frame positioning in the prediction layer. Experiments on the improved YOLOv5m algorithm demonstrate that the mean recognition accuracy of the method improves by 10.7 percentage points over the original method while maintaining the original model size and detection speed, and better solves the problems of small scale, dense arrangement and mutual occlusion of targets in mask wearing detection tasks in crowded places. © 2023 IEEE.