/face-mask-detector

The face mask detector is built using a convolutional neural network (CNN) to classify images of people as wearing masks or not wearing masks.

Primary LanguageJupyter NotebookMIT LicenseMIT

Face Mask Detector

This project is a part of the Graduate Rotational Internship Program (GRIP) offered by The Sparks Foundation. The goal of this project is to create a face mask detector using deep learning and computer vision techniques.

Overview

The face mask detector is built using a convolutional neural network (CNN) to classify images of people as wearing masks or not wearing masks. The dataset used for training and evaluating the model is obtained from Kaggle.

Dataset

The "Face Mask Detection" dataset can be found on Kaggle at the following link: Face Mask Detection Dataset. The dataset contains 12,000 images, which are divided into three sets: training, validation, and testing.

Dependencies

  • TensorFlow
  • NumPy
  • pandas
  • OpenCV
  • scikit-learn
  • Matplotlib

Usage

  1. Set up a Kaggle notebook and import the required datase.
  2. Install the necessary libraries and import them in your notebook.
  3. Load and preprocess the dataset, splitting it into training, validation, and test sets.
  4. Define and compile the CNN model.
  5. Train the model using the training dataset and validate it using the validation dataset.
  6. Evaluate the model using the test dataset.
  7. Save the model for future use.

License

This project is released under the MIT License.