NahidEbrahimian / Face-Mask-Detection

Real Time Face Mask Detection using Tensorflow/Keras | Inference on Image, Video and Webcam on ONNX Runtime Environment

Face-Mask-Detection

Real Time Face Mask Detection on Image and Video

• Model training on face-mask dataset using tensorflow and keras

• train.ipynb

• inference-img.py

• inference-webcam.py

• install.sh

• onnx inference

• inference-video

• docker file

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