/Mask-Detection

Django Mask Detection app based on computer vision and deep learning using OpenCV and Pytorch

Primary LanguagePython

Mask-Detection

Mask Detection app

Dataset: https://www.kaggle.com/datasets/andrewmvd/face-mask-detection

Example

Dependencies

  • Pytorch
  • OpenCV
  • Django, FastAPI (2 different branches)
  • Faster R-CNN ResNet 50

Setup

  1. Create virtual environment python -m venv enviro
  2. Activate the virtual environment.\enviro\Scripts\activate
  3. Install dependencies pip install -r requirements.txt

Usage

  1. Train the model (Remember to clone dataset and and set the parameters in config.py)
 python train.py
  1. Run the app (If Django)
 python manage.py runserver