Application of Deep Learning Face-mask Detection Based on YOLOv5
DOI: 10.23977/autml.2022.030302 | Downloads: 34 | Views: 752
Author(s)
Zixuan Xia 1, Ming Zhu 2
Affiliation(s)
1 School of Software Engineering, Xi'an Jiao Tong University, Xi'an, 710049, China
2 Department of Information Technology, Hunan Police Academy, Changsha, 410138, China
Corresponding Author
Zixuan XiaABSTRACT
Since the outbreak of coronavirus in 2019, people around the world have started to wear masks to avoid the further spread of the virus. In order to better control the epidemic, it is very necessary to supervise the wearing of masks. In this paper, a deep learning model based on yolov5 is established for mask recognition and detection, and the model is trained and tested through datasets. Finally, mask detection is carried out for people in images and videos using local computer equipment, and the best weight of training and the training accuracy of nearly 95% is obtained. This paper combines the algorithms of binary classification, convolutional neural network, and deep learning object detection to effectively and accurately train and test the model, which has a certain reference value.
KEYWORDS
Computer Vision, COVID-19, Deep Learning, Object Detection, YOLOv5CITE THIS PAPER
Zixuan Xia, Ming Zhu, Application of Deep Learning Face-mask Detection Based on YOLOv5. Automation and Machine Learning (2022) Vol. 3: 7-12. DOI: http://dx.doi.org/10.23977/autml.2022.030302.
REFERENCES
[1] Lehmann, E.Y., Lehmann, L.S. Responding to Patients Who Refuse to Wear Masks during the Covid-19 Pandemic. J Gen Intern Med 36, 2814–2815 (2021).
[2] Who coronavirus disease (covid-19) dashboard, https://covid19.who.int/, online accessed Aug 25, 2022
[3] Jingjing Nie, Linna Kang, Yaya Pian & Jihong Hu. The need for more robust research on the effectiveness of masks in preventing COVID-19 transmission. Published Online:19 Apr 2022
[4] Jiang, X.; Gao, T.; Zhu, Z.; Zhao, Y. Real-Time Face Mask Detection Method Based on YOLOv3. Electronics 2021, 10, 837.
[5] J. Ieamsaard, S. N. Charoensook and S. Yammen, "Deep Learning-based Face Mask Detection Using YoloV5," 2021 9th International Electrical Engineering Congress (iEECON), 2021, pp. 428-431,
[6] F. M. J. Mehedi Shamrat, S. Chakraborty, M. M. Billah, M. A. Jubair, M. S. Islam and R. Ranjan, "Face Mask Detection using Convolutional Neural Network (CNN) to reduce the spread of Covid-19," 2021 5th International Conference on Trends in Electronics and Informatics (ICOEI), 2021, pp. 1231-1237
[7] Li, Ye, Kangning Yin, Jie Liang, Chunyu Wang, and Guangqiang Yin. "A Multi-task Joint Framework for Real-time Person Search." arXiv preprint arXiv: 2012. 06418 (2020)
[8] Xu Renjie & Lin Haifeng & Lu Kangjie & Cao Lin & Liu Yunfei. (2021). A Forest Fire Detection System Based on Ensemble Learning. Forests. 12. 217. 10.3390/f12020217.
[9] Haghighi S, Jasemi M, Hessabi S, et al. PyCM: Multiclass confusion matrix library in Python [J]. Journal of Open Source Software, 2018, 3(25): 729.
[10] M. Heydarian, T. E. Doyle and R. Samavi, "MLCM: Multi-Label Confusion Matrix," in IEEE Access, vol. 10, pp. 19083-19095, 2022.
Downloads: | 1745 |
---|---|
Visits: | 71303 |
Sponsors, Associates, and Links
-
Power Systems Computation
-
Internet of Things (IoT) and Engineering Applications
-
Computing, Performance and Communication Systems
-
Journal of Artificial Intelligence Practice
-
Advances in Computer, Signals and Systems
-
Journal of Network Computing and Applications
-
Journal of Web Systems and Applications
-
Journal of Electrotechnology, Electrical Engineering and Management
-
Journal of Wireless Sensors and Sensor Networks
-
Journal of Image Processing Theory and Applications
-
Mobile Computing and Networking
-
Vehicle Power and Propulsion
-
Frontiers in Computer Vision and Pattern Recognition
-
Knowledge Discovery and Data Mining Letters
-
Big Data Analysis and Cloud Computing
-
Electrical Insulation and Dielectrics
-
Crypto and Information Security
-
Journal of Neural Information Processing
-
Collaborative and Social Computing
-
International Journal of Network and Communication Technology
-
File and Storage Technologies
-
Frontiers in Genetic and Evolutionary Computation
-
Optical Network Design and Modeling
-
Journal of Virtual Reality and Artificial Intelligence
-
Natural Language Processing and Speech Recognition
-
Journal of High-Voltage
-
Programming Languages and Operating Systems
-
Visual Communications and Image Processing
-
Journal of Systems Analysis and Integration
-
Knowledge Representation and Automated Reasoning
-
Review of Information Display Techniques
-
Data and Knowledge Engineering
-
Journal of Database Systems
-
Journal of Cluster and Grid Computing
-
Cloud and Service-Oriented Computing
-
Journal of Networking, Architecture and Storage
-
Journal of Software Engineering and Metrics
-
Visualization Techniques
-
Journal of Parallel and Distributed Processing
-
Journal of Modeling, Analysis and Simulation
-
Journal of Privacy, Trust and Security
-
Journal of Cognitive Informatics and Cognitive Computing
-
Lecture Notes on Wireless Networks and Communications
-
International Journal of Computer and Communications Security
-
Journal of Multimedia Techniques
-
Computational Linguistics Letters
-
Journal of Computer Architecture and Design
-
Journal of Ubiquitous and Future Networks