Cargando…

Face mask detection and classification via deep transfer learning

Wearing a mask is an important way of preventing COVID-19 transmission and infection. German researchers found that wearing masks can effectively reduce the infection rate of COVID-19 by 40%. However, the detection of face mask-wearing in the real world is affected by factors such as light, occlusio...

Descripción completa

Detalles Bibliográficos
Autores principales: Su, Xueping, Gao, Meng, Ren, Jie, Li, Yunhong, Dong, Mian, Liu, Xi
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Springer US 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8656443/
https://www.ncbi.nlm.nih.gov/pubmed/34903950
http://dx.doi.org/10.1007/s11042-021-11772-5
_version_ 1784612282477576192
author Su, Xueping
Gao, Meng
Ren, Jie
Li, Yunhong
Dong, Mian
Liu, Xi
author_facet Su, Xueping
Gao, Meng
Ren, Jie
Li, Yunhong
Dong, Mian
Liu, Xi
author_sort Su, Xueping
collection PubMed
description Wearing a mask is an important way of preventing COVID-19 transmission and infection. German researchers found that wearing masks can effectively reduce the infection rate of COVID-19 by 40%. However, the detection of face mask-wearing in the real world is affected by factors such as light, occlusion, and multi-object. The detection effect is poor, and the wearing of cotton masks, sponge masks, scarves and other items greatly reduces the personal protection effect. Therefore, this paper proposes a new algorithm for mask detection and classification that fuses transfer learning and deep learning. Firstly, this paper proposes a new algorithm for face mask detection that integrates transfer learning and Efficient-Yolov3, using EfficientNet as the backbone feature extraction network, and choosing CIoU as the loss function to reduce the number of network parameters and improve the accuracy of mask detection. Secondly, this paper divides the mask into two categories of qualified masks (N95 masks, disposable medical masks) and unqualified masks (cotton masks, sponge masks, scarves, etc.), creates a mask classification data set, and proposes a new mask classification algorithm that the combines transfer learning and MobileNet, enhances the generalization of the model and solves the problem of small data size and easy overfitting. Experiments on the public face mask detection data set show that the proposed algorithm has a better performance than existing algorithms. In addition, experiments are performed on the created mask classification data set. The mask classification accuracy of the proposed algorithm is 97.84%, which is better than other algorithms.
format Online
Article
Text
id pubmed-8656443
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher Springer US
record_format MEDLINE/PubMed
spelling pubmed-86564432021-12-09 Face mask detection and classification via deep transfer learning Su, Xueping Gao, Meng Ren, Jie Li, Yunhong Dong, Mian Liu, Xi Multimed Tools Appl Article Wearing a mask is an important way of preventing COVID-19 transmission and infection. German researchers found that wearing masks can effectively reduce the infection rate of COVID-19 by 40%. However, the detection of face mask-wearing in the real world is affected by factors such as light, occlusion, and multi-object. The detection effect is poor, and the wearing of cotton masks, sponge masks, scarves and other items greatly reduces the personal protection effect. Therefore, this paper proposes a new algorithm for mask detection and classification that fuses transfer learning and deep learning. Firstly, this paper proposes a new algorithm for face mask detection that integrates transfer learning and Efficient-Yolov3, using EfficientNet as the backbone feature extraction network, and choosing CIoU as the loss function to reduce the number of network parameters and improve the accuracy of mask detection. Secondly, this paper divides the mask into two categories of qualified masks (N95 masks, disposable medical masks) and unqualified masks (cotton masks, sponge masks, scarves, etc.), creates a mask classification data set, and proposes a new mask classification algorithm that the combines transfer learning and MobileNet, enhances the generalization of the model and solves the problem of small data size and easy overfitting. Experiments on the public face mask detection data set show that the proposed algorithm has a better performance than existing algorithms. In addition, experiments are performed on the created mask classification data set. The mask classification accuracy of the proposed algorithm is 97.84%, which is better than other algorithms. Springer US 2021-12-09 2022 /pmc/articles/PMC8656443/ /pubmed/34903950 http://dx.doi.org/10.1007/s11042-021-11772-5 Text en © The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2021 This article is made available via the PMC Open Access Subset for unrestricted research re-use and secondary analysis in any form or by any means with acknowledgement of the original source. These permissions are granted for the duration of the World Health Organization (WHO) declaration of COVID-19 as a global pandemic.
spellingShingle Article
Su, Xueping
Gao, Meng
Ren, Jie
Li, Yunhong
Dong, Mian
Liu, Xi
Face mask detection and classification via deep transfer learning
title Face mask detection and classification via deep transfer learning
title_full Face mask detection and classification via deep transfer learning
title_fullStr Face mask detection and classification via deep transfer learning
title_full_unstemmed Face mask detection and classification via deep transfer learning
title_short Face mask detection and classification via deep transfer learning
title_sort face mask detection and classification via deep transfer learning
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8656443/
https://www.ncbi.nlm.nih.gov/pubmed/34903950
http://dx.doi.org/10.1007/s11042-021-11772-5
work_keys_str_mv AT suxueping facemaskdetectionandclassificationviadeeptransferlearning
AT gaomeng facemaskdetectionandclassificationviadeeptransferlearning
AT renjie facemaskdetectionandclassificationviadeeptransferlearning
AT liyunhong facemaskdetectionandclassificationviadeeptransferlearning
AT dongmian facemaskdetectionandclassificationviadeeptransferlearning
AT liuxi facemaskdetectionandclassificationviadeeptransferlearning