README.md

COVID-19 Face Mask Detector

Reproduction

git clone [email protected]:JadHADDAD92/covid-mask-detector.git
cd covid-mask-detector

# Download dataset and export it to pandas DataFrame
python -m covid-mask-detector.data_preparation

Training

python -m covid-mask-detector.train

Testing on videos

python -m covid-mask-detector.video modelPath videoPath

Usage

Usage: video.py [OPTIONS] MODELPATH VIDEOPATH

  modelPath: path to model.ckpt

  videoPath: path to video file to annotate

Options:
  --output PATH  specify output path to save video with annotations

Pretrained model

covid-mask-detector/models/face_mask.ckpt

Check Medium post: How I built a Face Mask Detector for COVID-19 using PyTorch Lightning