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.
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
- π‘ 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.
- Python 3.9+
- TensorFlow / Keras
- VGGFace (via
keras-vggface) - Google Colab
- NumPy, Matplotlib, shutil, urllib, zipfile
-
Data Download & Preprocessing
Downloads celebrity face images from a Kaggle-hosted ZIP file, extracts them, and prepares them for training. -
Embedding Extraction
Uses VGGFace (pre-trained on millions of faces) to extract facial embeddings. -
Model Training
Embeddings are used to train a classifier (e.g., SVM or KNN) to identify which celebrity is in the image. -
Prediction
Given a new input image, the model predicts which celebrity the face most closely resembles.
- π― 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.
Celebrity_Face_Recognition_VGGFace_Model/
β
βββ Celebrity_Face_Recognition_VGGFace_Model.ipynb # Main Google Colab notebook
βββ README.md # Project documentation
βββ requirements.txt # Dependencies (optional)