Skip to content

Built a face recognition system using the VGGFace model to identify celebrity faces. Implemented in Google Colab with automatic dataset handling, face embedding, and classification. Showcases skills in transfer learning, computer vision, and deep learning.

Notifications You must be signed in to change notification settings

manyagautam/Celebrity-Face-Recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

3 Commits
Β 
Β 
Β 
Β 

Repository files navigation

Celebrity Face Recognition using VGGFace Model

Welcome to the Celebrity Face Recognition project! This repository demonstrates a deep learning-based face recognition system using the VGGFace model. The system identifies and verifies celebrity faces from a dataset using pre-trained facial embeddings.


πŸ” Project Overview

This project is designed to detect and recognize celebrity faces from images using transfer learning and the VGGFace model architecture. Built and tested in Google Colab, it automates the process from dataset loading and preprocessing to model evaluation and prediction.

βœ… Live Environment: Built and tested in Google Colab using TensorFlow/Keras
πŸ“ Dataset: Kaggle - Celebrity Faces Dataset


πŸš€ Features

  • πŸ’‘ Deep Learning Powered: Uses VGGFace (based on VGG16) for accurate face embeddings.
  • πŸ“¦ Dataset Handling: Automatically fetches, extracts, and prepares the celebrity face dataset.
  • πŸ“Έ Face Embedding: Generates embeddings to compare and classify faces.
  • 🧠 Recognition Pipeline: Face detection β†’ Embedding β†’ Classification.
  • πŸ“Š Highly Modular: Cleanly structured code to plug-and-play different models or datasets.
  • πŸ“ˆ Scalable: Suitable for extending to real-time recognition or custom datasets.

πŸ› οΈ Tech Stack

  • Python 3.9+
  • TensorFlow / Keras
  • VGGFace (via keras-vggface)
  • Google Colab
  • NumPy, Matplotlib, shutil, urllib, zipfile

πŸ§ͺ How It Works

  1. Data Download & Preprocessing
    Downloads celebrity face images from a Kaggle-hosted ZIP file, extracts them, and prepares them for training.

  2. Embedding Extraction
    Uses VGGFace (pre-trained on millions of faces) to extract facial embeddings.

  3. Model Training
    Embeddings are used to train a classifier (e.g., SVM or KNN) to identify which celebrity is in the image.

  4. Prediction
    Given a new input image, the model predicts which celebrity the face most closely resembles.


πŸ“Œ Project Highlights (Why It’s Resume-Worthy)

  • 🎯 Demonstrates practical application of transfer learning in facial recognition.
  • πŸ” Hands-on with real-world data preprocessing and automation.
  • πŸ€– Showcases skills in computer vision, neural networks, and data engineering.
  • βœ… Built for scalability and production-ready adaptation.

πŸ“‚ Repository Structure

Celebrity_Face_Recognition_VGGFace_Model/
β”‚
β”œβ”€β”€ Celebrity_Face_Recognition_VGGFace_Model.ipynb  # Main Google Colab notebook
β”œβ”€β”€ README.md                                        # Project documentation
└── requirements.txt                                 # Dependencies (optional)

About

Built a face recognition system using the VGGFace model to identify celebrity faces. Implemented in Google Colab with automatic dataset handling, face embedding, and classification. Showcases skills in transfer learning, computer vision, and deep learning.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages