A deep learning based model that detects whether a person has a face mask on or not with nearly 100% accuracy.
Author : Ashay Ajbani
Pretrained model used : MobileNetV2
Trainable parameters : 3M
Number of classes : 2
Train Accuracy : 99.29%
Validation Accuracy : 100%
- For training refer mask_detection.ipynb
- For testing on your own image run predict_image.py
Techniques used to reduce overfitting
- Data Augmentation
- BatchNormalization
- Dropout