A Mask Detection Algorithm Based on Improved Yolov5s

Abstract

Aiming at the severe form of new coronavirus epidemic prevention and control, a target detection algorithm is proposed to detect whether masks are worn in public places. The Ghostnet and SElayer modules with fewer design parameters replace the BottleneckCSP part in the original Yolov5s network, which reduces the computational complexity of the model and improves the detection accuracy. The bounding box regression loss function DIOU is optimized, the DGIOU loss function is used for bounding box regression, and the center coordinate distance between the two bounding boxes is considered to achieve a better convergence effect. In the feature pyramid, the depthwise separable convolution DW is used to replace the ordinary convolution, which further reduces the amount of parameters and reduces the loss of feature information caused by multiple convolutions. The experimental results show that compared with the yolov5s algorithm, the proposed method improves the mAP by 4.6% and the detection rate by 10.7 frame/s in the mask wearing detection. Compared with other mainstream algorithms, the improved yolov5s algorithm has better generalization ability and practicability.