/face_mask_detector_pytorch

Primary LanguagePython

Face Mask Detection

Model developed using Pytorch Lightning

Image

Model file is model_w.ckpt

How to run:

  1. Clone this repo
  2. Run requirements.txt
  3. Run main.py

How to train:

  1. Clone this repo
  2. Run trainer.py

Todo:

  1. Enhance the prediction capability