This project involves building a face detection model using Convolutional Neural Networks (CNN) with Keras. The model is designed to classify images into two classes: whether a person is wearing a mask or not. The architecture consists of multiple convolutional layers followed by dense layers, with dropout layers to prevent overfitting.
https://www.kaggle.com/datasets/omkargurav/face-mask-dataset
To run this model, ensure you have the following dependencies installed:
pip install tensorflow
pip install numpy
- input_shape: The shape of the input images (default is
(128, 128, 3)
). - num_of_class: The number of classes for classification (default is
2
for binary classification).
- Clone the repository.
- Install the required dependencies.
- Load your dataset and preprocess it.
- Define and compile the model.
- Train the model on your dataset.
- Evaluate and make predictions with the model.