🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀
-
This project is based on https://github.com/chandrikadeb7/Face-Mask-Detection , but with 3 classes(with_mask,without_mask,incorrect_mask)
-
Project is using Tensorflow, keras to train model with format HDF5, and use flask framework to show the real time Face Mask Recognition
-
In this project, I'm using transfer learning to train model with pretrained-model is VGG-16
-
This project using Python 3.8.10 and GPU: GTX-1650TI with CUDA support: v11.2 and CUDNN: v8.1.1
- So firstly, you have to download the project and then install requirement.txt following this command line:
pip install -r requirement.txt
-
So the code training and demo in folder source: Run "python Training.py" to train your own model with your own dataset Run "python demo.py" to inference your result or my result model with webcam
-
The code for web inference in the folder: Flask Run "python app.py" to inference in your local host
My model have a good accuracy: 0.9889 with 3 classes!!!!!
So good luck and enjoy the result
You can use my model with the following link here: https://drive.google.com/drive/folders/1iFvdBTWkNwh-umId8gAcvEcicwu2LrcJ?usp=sharing
Download my pretrained-model and path to ModelResult/data/datasets
Thanks 🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀🚀