Face Recognition of Top 3 Khans of Bollywood

A Deep Learning App to identify Top 3 Khans (Aamair Khan, Salman Khan, Shahrukh Khan) of Bollywood Industry

Usage / Installation

This section will guide you through a series of steps to setup this project on your computer.

Requirements

Make sure you have the following installed to proceed further into the installation process

• Python 3.5 (or above)
• Python Package Manager - ( PIP )
• Git

Install Dependencies for the project

All the dependencies and Packages have been conveniently bundles into a 'requirements.txt' file for a smooth installation and have the Project Up and running on your machine with minimal effort and time. Run the following commands from the terminal .

Get a complete copy of the project by cloning into a suitable directory

git clone https://github.com/akhilgup/face_recognition.git

Install all the dependencies by running the command from the terminal.

sudo pip install -r requirements.txt

Usage

• For training and subsequently generating the model, run the training.py file by the following command.

python training.py

This generates a model file face_reg.h5 which is then used to give predictions.

• For testing an image for giving predictions for the parameters Aamair Khan, Salman Khan, Shahrukh Khan, place your image file in the same directory with the name test2.jpg and run the testing.py file by the following command

python testing.py

Technical Details

Training

• The model is a result of Convolutional Neural Network with 2 Convolution Layers, 2 Pooling Layers applied with the Rectified Linear Unit (ReLU) activation function with Filter Size of 5 along with Flatten Layer (Dense) made up of 500 neurons using Rectified Linear Unit (ReLU) as it's activation function.

Training Parameters :-

EPOCHS = 100
INIT_LR = 1e-3 // Initial Learning Rate
BATCH_SIZE = 32

• Output Layer is Dense Layer consisting of neurons equal to the no. of Target Classes 3 using softmax as it's activation function.

Image Pre-Processing

• All training images were scaled to 64 x 64 in Grayscale and all the pixels were feature scaled by dividing each pixel by 255.0.
• Image generator function has been used to increase the training data by flipping, rotating images : Keras Image PreProcessing .

Results

Training Accuracy : 91 %

Validation Accuracy: 90.7 %

Training Loss : 0.2414

Validation Loss : 0.2673

Libraries / Frameworks Used

• Deep Learning Libraries
  • TensorFlow
  • Keras
  • h5py
• Machine Learning Libraries
  • SciKit Learn
  • Pandas
  • SciPy
  • Numpy
  • MatPlotLib
  • Seaborn
• Computer Vision Libraries
  • OpenCV