Face-Mask-Detection
Face Mask Detection on Image and Video using tensorflow and keras
-
Train Neural Network on face-mask dataset using tensorflow and keras
-
train.ipynb
-
inference-img.py
-
inference-webcam.py
Dataset:
Dataset link: face-mask-dataset
Inference:
for inference on image, put your images in ./input
directory and run the following command. output images will be saved on output
folder.
python3 inference-img.py --input input/01.jpeg
for inference on video using webcam, run the following command:
python3 inference-webcam.py
Useful links:
Face-Alignment preprocessing used in the inference step: https://github.com/SajjadAemmi/Face-Alignment