Skip to content

Latest commit

 

History

History
33 lines (22 loc) · 682 Bytes

README.md

File metadata and controls

33 lines (22 loc) · 682 Bytes

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