A CNN based facial keypoints detector.
Project description
Facial Keypoints Detection
A computer vision project to build a facial keypoints detection system.
Table of Contents
Project Overview
- Facial keypoints detection system has variety of applications, including:
- Facial tracking.
- Facial pose recognition.
- Facial filters.
- Emotion recognition.
- Medical diagnosis: Identifying dysmorphic facial symptoms.
- Detecting facial keypoints is a challenging problem given the variations in both facial features as well as image conditions. Facial features may differ according to size, position, pose and expression, while image qualtiy may vary with illumination and viewing angle.
- In this project, Convolutional Neural Network (CNN) based facial keypoints detector system has been implemented to detect 68 facial keypoints (also called facial landmarks) around important areas of the face: the eyes, corners of the mouth, the nose, etc. using computer vision techniques and deep learning architectures.
- The project is broken up into a few main parts in 4 Python notebooks:
- Notebook 1: Loading and Visualizing the Facial Keypoint Data.
- Notebook 2: Defining and Training a Convolutional Neural Network (CNN) to Predict Facial Keypoints.
- Notebook 3: Facial Keypoint Detection Using Haar Cascades and a Trained CNN.
- Notebook 4: Applications - Facial filters, Face Blur.
- The implemented Python package code is facial_keypoints_detecter.
Data Description
- Facial keypoints are the small magenta dots shown on each of the faces in the image above.
- In each training and test image, there is a single face and 68 keypoints, with coordinates (x, y), for that face.
- These keypoints mark important areas of the face: the eyes, corners of the mouth, the nose, etc. Here they are, numbered, and you can see that specific ranges of points match different portions of the face.
Training and Testing Data
- This facial keypoints dataset consists of 5770 color images.
- 3462 are training images.
- 2308 are test images.
- The information about the images and keypoints in this dataset are summarized in CSV files.
Original + Augmented data
Note: Datasets are explored in Notebook 1.
Note: This set of image data has been extracted from the YouTube Faces Dataset, which includes videos of people in YouTube videos. These videos have been fed through some processing steps and turned into sets of image frames containing one face and the associated keypoints.
mkdir data
wget -P data/ https://s3.amazonaws.com/video.udacity-data.com/topher/2018/May/5aea1b91_train-test-data/train-test-data.zip
unzip -n data/train-test-data.zip -d data
Results
- A custom made Python package
facial_keypoints_detecter
which contains a classifier, plotting & feature extraction functionalities, and datasets for the project. - The trained model has been implemented for 2 example applications:
- Facial filters
- Face blur
Facial filters
Face blur
Feature visualization
- The project results and package usage have been clearly demonstrated in
Python package facial_keypoints_detecter
- This project utilizes a custom-made package
facial_keypoints_detecter
which contains a classifier, plotting & feature extraction functionalities, and datasets for the project. - Libraries used:
Python 3
,PyTorch
,torchvision
,OpenCV-Python
,Matplotlib
,pandas
,numpy
. - This library contains a cnn model, pre-processing tools, plotting tools, and datasets loading tools for this project.
- facial_keypoints_detecter contains a cnn model, pre-processing
- Main libraries used: PyTorch, OpenCV-Python, matplotlib, pandas, numpy.
Dependencies
Python 3
, PyTorch
, torchvision
, OpenCV-Python
, Matplotlib
, pandas
, numpy
.
Installation
# Install package from PyPI >>
pip install facial_keypoints_detecter
# or
# Install package from GitHub >>
pip install git+https://github.com/ShashankKumbhare/facial-keypoints-detecter.git#egg=facial-keypoints-detecter
Package usage has benn demonstrated in Notebook 1, Notebook 2, Notebook 3, Notebook 4.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file facial_keypoints_detecter-1.0.0.tar.gz
.
File metadata
- Download URL: facial_keypoints_detecter-1.0.0.tar.gz
- Upload date:
- Size: 14.9 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | f914c7e970d57a666a62fa4a1b1dc17c6ac140aa32f776a3a12f961cb68f3eea |
|
MD5 | 685a9143f9cdc0c82e975608e367f5dc |
|
BLAKE2b-256 | 51327208591c6c774b1ea81025ac20e58f38fd129c1ee3e88769ca7b4001ab37 |
File details
Details for the file facial_keypoints_detecter-1.0.0-py3-none-any.whl
.
File metadata
- Download URL: facial_keypoints_detecter-1.0.0-py3-none-any.whl
- Upload date:
- Size: 15.0 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c3702a42a3a0b170b769940e8e2b6b153a8b6799cce8b0524362232738b54d69 |
|
MD5 | 111b3507387ff0fefdb87bc1ab71ad21 |
|
BLAKE2b-256 | 057e6761b329baafcef8f887296442de5af6a949ec7ac7974890f145e35fb64d |