Skip to content

Understanding facial beauty with deep learning.

License

Unknown, Unknown licenses found

Licenses found

Unknown
LICENSE.md
Unknown
LICENSE.txt
Notifications You must be signed in to change notification settings

erogol/beauty.torch

Repository files navigation

beauty.torch

This project serves a deep learning model scoring selfie images between 1 to 10 based on image and face attributes. You can learn the technical details of this project from this blog post. Use [resnet.torch] (https://github.com/erogol/resnet.torch), if you plan to follow all the training pipeline described on the post.

Given image is processed as follows;

  1. Detect face.
  2. Find landmarks
  3. Rotate image to align face.
  4. Fill gaps with constant pixel value.
  5. Send into scoring model.

For an example use check notebook ExampleUse.ipynb

Dataset & Converged Final Model

  • Contact me from erengolge at gmail.com

Requirements

Main requirement is Torch computing framework.

Models

dlib face model - place under utils/
beauty model - GPU model (use utils/convert2cpu.lua for setting it for CPU) place under trained/
optimstate - if you like to fine-tune the model.

Python

dlib sudo pip install dlib - face and landmark detection)
lutorpy sudo pip install lutorpy - using torch model on python
skimage sudo pip install skimage - image processing
cv2 sudo pip install cv2 - OpenCV python module

What you have here useful

  • Face alignment code in utils/img_processing.py.
  • A template for porting Torch models to python in utils/Classifier.py.
  • The model itself

Examples

Attention of the trained model.
alt text

Sorting A. Lima images from Google Search.
alt tag

About

Understanding facial beauty with deep learning.

Topics

Resources

License

Unknown, Unknown licenses found

Licenses found

Unknown
LICENSE.md
Unknown
LICENSE.txt

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published