Utilizing Transfer Learning, the system works on detecting a face mask in real-time. The model is trained on a balanced dataset of 7,500 samples on a pre-trained MobileNetV2.
- Python
- OpenCV
- MobileNetV2 (Pre-trained model)
- TensorFlow
- Keras
- NumPy
- sklearn
- matplotlib
- Detects for a face mask
- Can detect multiple faces in a single frame
- Has an accuracy of about 98%
Type python trainMaskDetector.py
in command prompt.
Type python maskDetector.py
in command prompt.