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 link: face-mask-dataset

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