Facial Image Classification

In this project, one or more human faces are detected through images and name predictions are made for those faces detected by AI models trained in the background.


Working Demonstration

I have used Haar Cascade Classifier and Wavelet Transform to detect faces in the images. Then after preprocessing the datasets, I tested my model on various classifiers(SVM, Logistic Regression,Random Forest). Then I have used the best performing model to predict the names of the faces detected in the images.


Usage

To run this project, you need to have Python 3.6 or above installed on your system. Then you can clone this repository and install the required libraries using the following command [/Model directory]:

pip install -r requirements.txt

Then you can activate the flask server using the following command [/Server directory]:

python server.py

Then you can run the application by opening the index.html file in the browser [/UI directory].


Features

  • Detect faces in images
  • User friendly UI
  • Predict names of the faces detected
  • Deploy the model on the web

License

This project is licensed under the Apache License - see the LICENSE file for details. HyperWrite Logo.