Skip to content

JadHADDAD92/covid-mask-detector

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Open in Visual Studio Code license CodeQL

Buy Me A Coffee

COVID-19 Face Mask Detector

samples

Reproduction

git clone git@github.com:JadHADDAD92/covid-mask-detector.git
# Or
git clone https://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