/Mask-Detector

ML Notebook to Detect mask in frame of a image which trains a CNN model using transfer learning in which mobilenetv2 model architecture is used with imagenet weights as a base model,on top which a head model is made using different layers like average pooling and dense layers to find specific patterns in the images of mask and without mask.

Primary LanguageJupyter Notebook

ML notebook which trains a CNN model using transfer learning in which mobilenetv2 model architecture is used with imagenet weights as a base model,on top which a head model is made using different layers like average pooling and dense layers to find specific patterns in the images of mask and without mask.

Accuracy of 98% on test data

Dataset Collection-

RMFD

Kaggle

Google Search

Testing of model on out of dataset images- test