Skip to content

Using MTCNN and VggFace, created a webApp which tells your look alike Bollywood celebrity

Notifications You must be signed in to change notification settings

EphronM/Celebrity_face_match

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Finding your look alike celebrity

Python Framework Frontend Dataset

We all are very curious about comparing ourselfs to other. Imagine comparing your facial features with Bollywood celebrities to find your lookalike celeb.

Using MTCNN and VggFACE, created a webApp where you could upload your image and the model tells you to which Bollywood celeb your face lookalike to.

homepage

Local Deployment

Clone the repository

git clone https://github.com/EphronM/Celebrity_face_match.git

Download the dataset from kaggle

image

  • Combine all the folders into a single folder named data

Create a conda environment after opening the repository

conda create -n face_env python=3.9 -y
conda activate face_env
  • Installing the required dependencies
pip install -r requirements.txt

Creating artifacts

  • Croping celeb face and saving it in data cropped
  python src\01_genreating_image_pickle.py
  • Extracting features from all the images and saving the pickle file
  python src\02_feature_extraction.py

We are using MTCNN as the face detector and VggFace prediction model to extract the feature similarities. Finally using Cosine Similarity, we find the similar celeb face.

All set to deploy

  streamlit run app.py

Model performence

Lets give the model few real bollywood celeb look alikes peoples photos and see,

result1

Works well with David Saharia who is real life tiger shroff lookalike

Lets try it with the Amitabh Bachchan transformed look.

Captaure

worked good with 75% match.

result2 Capture

This all about this project.

Happy coding Focks!!

Author: EphronM
Email: ephronmartin2016@gmail.com

About

Using MTCNN and VggFace, created a webApp which tells your look alike Bollywood celebrity

Topics

Resources

Stars

Watchers

Forks