The official code for "Wearing face mask detection using deep learning through COVID-19 pandemic_".
@article{Faculty of Mechanical Engineering, Tarbiat Modares University, Tehran, Iran,
title={Wearing face mask detection using deep learning through COVID-19 pandemic},
author={Javad Khoramdela
, Soheila Hatamib
, and Majid Sadedel},
journal={https://arxiv.org/abs/2305.00068},
year={2023}
}
first download models and save them in same directory with IPYNB file as jupyter notebooks one of them is for YOLOv4 and the other is for Mobilnetv2 and the odel is same as nootbooks.
Google Drive Link:().
-
Download the face mask detection dataset from here.
-
- Run the following code to install the Requirements.
pip install -r requirements.txt
-
Run the below code to train the Wearing-face-mask-detection on the face mask detection dataset. SSD.ipynb YOLOv4.ipynb
-
Test trained model with this dataset in in IPYNB too.
Performance comparision on face-mask-detection dataset.