FMD-Yolo: An efficient face mask detection method for COVID-19 prevention and control in public

被引:99
作者
Wu, Peishu [1 ]
Li, Han [1 ]
Zeng, Nianyin [1 ]
Li, Fengping [2 ]
机构
[1] Xiamen Univ, Dept Instrumental & Elect Engn, Xiamen 361005, Fujian, Peoples R China
[2] Wenzhou Univ, Inst Laser & Optoelect Intelligent Mfg, Wenzhou 325035, Peoples R China
基金
中国国家自然科学基金;
关键词
Face mask detection; COVID-19; Improved YoloV3 algorithm; Feature extraction and fusion;
D O I
10.1016/j.imavis.2021.104341
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Coronavirus disease 2019 (COVID-19) is a world-wide epidemic and efficient prevention and control of this disease has become the focus of global scientific communities. In this paper, a novel face mask detection framework FMDYolo is proposed to monitor whether people wear masks in a right way in public, which is an effective way to block the virus transmission. In particular, the feature extractor employs Im-Res2Net-101 which combines Res2Net module and deep residual network, where utilization of hierarchical convolutional structure, deformable convolution and non-local mechanisms enables thorough information extraction from the input. Afterwards, an enhanced path aggregation network En-PAN is applied for feature fusion, where high-level semantic information and low-level details are sufficiently merged so that the model robustness and generalization ability can be enhanced. Moreover, localization loss is designed and adopted in model training phase, and Matrix NMS method is used in the inference stage to improve the detection efficiency and accuracy. Benchmark evaluation is performed on two public databases with the results compared with other eight state-of-the-art detection algorithms. At IoU = 0.5 level, proposed FMD-Yolo has achieved the best precision AP50 of 92.0% and 88.4% on the two datasets, and AP75 at IoU = 0.75 has improved 5.5% and 3.9% respectively compared with the second one, which demonstrates the superiority of FMD-Yolo in face mask detection with both theoretical values and practical significance. (c) 2021 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 45 条
[1]  
[Anonymous], 2015, P 3 INT C LEARN REPR
[2]   Evaluation of indoor hospital acclimatization of body temperature before COVID-19 fever screening [J].
Bassi, A. ;
Henry, B. M. ;
Pighi, L. ;
Leone, L. ;
Lippi, G. .
JOURNAL OF HOSPITAL INFECTION, 2021, 112 :127-128
[3]  
Bazarevsky V., 2019, ARXIV PREPRINT ARXIV
[4]  
Bochkovskiy A, 2020, ARXIV, DOI 10.48550/ARXIV.2004.10934
[5]  
Chen Rung-Ching, 2019, IMAGE VISION COMPUT, V87, P47, DOI DOI 10.1016/j.imavis.2019.04.007
[6]  
Chinrungrueng C., 1991, IJCNN 91 SEATTLE INT, V1, P855
[7]   Res2Net: A New Multi-Scale Backbone Architecture [J].
Gao, Shang-Hua ;
Cheng, Ming-Ming ;
Zhao, Kai ;
Zhang, Xin-Yu ;
Yang, Ming-Hsuan ;
Torr, Philip .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2021, 43 (02) :652-662
[8]  
Ghiasi G, 2018, ADV NEUR IN, V31
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]  
He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]