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