Skip to content

thomaslin1990/covid-mask-detector

 
 

Repository files navigation

COVID-19 Face Mask Detector

samples

Reproduction

git clone git@github.com: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

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%