This is an image classification problem, and the aim is to perform face mask detection using the Convolutional Neural Network (CNN) workflow in TensorFlow and OpenCV
- The dataset contains
train
(with 1315 images) andtest folders
(with 194 images), and each folder contains 2 subfolders (i.e., with_mask and without_mask), each having equal number of images for both train and test sets. Awith_mask
folder consists of images of people with mask, and the images of people without masks is stored in awithout_mask
folder.
This repo contains 6 files namely:
face_mask_detection2.ipynb
: Python file for model developmentimplement_mask_detection.py
: Python file for face mask detectionfacemaskdetectormodel2.h5
: CNN modelhaarcascade_frontalface_default.xml
: Frontal face detector