An Improved Lightweight YOLOv5 Model Based on Attention Mechanism for Face Mask Detection

by   Sheng Xu, et al.

Coronavirus 2019 has brought severe challenges to social stability and public health worldwide. One effective way of curbing the epidemic is to require people to wear masks in public places and monitor mask-wearing states by utilizing suitable automatic detectors. However, existing deep learning based models struggle to simultaneously achieve the requirements of both high precision and real-time performance. To solve this problem, we propose an improved lightweight face mask detector based on YOLOv5, which can achieve an excellent balance of precision and speed. Firstly, a novel backbone ShuffleCANet that combines ShuffleNetV2 network with Coordinate Attention mechanism is proposed as the backbone. Afterwards, an efficient path aggression network BiFPN is applied as the feature fusion neck. Furthermore, the localization loss is replaced with alpha-CIoU in model training phase to obtain higher-quality anchors. Some valuable strategies such as data augmentation, adaptive image scaling, and anchor cluster operation are also utilized. Experimental results on AIZOO face mask dataset show the superiority of the proposed model. Compared with the original YOLOv5, the proposed model increases the inference speed by 28.3 achieves the best mean average precision of 95.2 existing models, which is 4.4


Research on Mask Wearing Detection of Natural Population Based on Improved YOLOv4

Recently, the domestic COVID-19 epidemic situation has been serious, but...

Development of a face mask detection pipeline for mask-wearing monitoring in the era of the COVID-19 pandemic: A modular approach

During the SARS-Cov-2 pandemic, mask-wearing became an effective tool to...

RetinaMask: A Face Mask detector

Coronavirus disease 2019 has affected the world seriously, because peopl...

YOLOv5s-GTB: light-weighted and improved YOLOv5s for bridge crack detection

In response to the situation that the conventional bridge crack manual d...

Lightweight wood panel defect detection method incorporating attention mechanism and feature fusion network

In recent years, deep learning has made significant progress in wood pan...

Light-YOLOv5: A Lightweight Algorithm for Improved YOLOv5 in Complex Fire Scenarios

In response to the existing object detection algorithms are applied to c...

Learning to mask: Towards generalized face forgery detection

Generalizability to unseen forgery types is crucial for face forgery det...

Please sign up or login with your details

Forgot password? Click here to reset