Hi!, welcome to our repository. This is a fun project developed to get better insights on Computer Vision and Image Classification. We developed MobileNetV2 based model to identify if a person is wearing a mask or not. It is a simple web application developed with the help of Flask. The model is binary classification type model with classes respectively as With Mask or Without Mask. The dataset that we used for training the model can be found here
These instructions will get you a copy of the project up and running on your local machine for development and testing purposes.
Before initializing with this project you must know the following concepts:
- Machine Learning -
Classification - Deep Learning -
Neural Networks,CNN Networks,TensorFlow(TF2) - Transfer Learning
- Computer Vision -
OpenCV Library - Web Application Development -
Flask,Jinja Templates
-
Clone the repository
git clone https://github.com/Purnay087/FaceMaskDetection-TF2-Flask.git. -
Install requirements using
pip install -r requirements.txt
If you want a custom dataset for your model follow these steps:-
- Download your dataset
- Categorize your dataset in two folders named with_mask and without_mask
- Put these two folders inside dataset folder inside project's home directory
- Use this file to train your model
Execute main.py file to start the web application
