kingvision / FacialLandmark_Caffe

Facial Landmark Detection with Caffe CNN. Face alignment avaiable.

What's this

1. Facial Landmark Detection with Caffe CNN.
2. Implementation of the two nets from different paper.

• Net1: Approaching human level facial landmark localization by deep learning
• Net2: TCNN: Facial Landmark Detection with Tweaked Convolutional Neural Networks

How to prepare dataset

Just enter dataset folder and run script to get training data

cd dataset
python get_dataset.py

How to training

Please run caffe command in the root of this project.

• train net1

caffe train --solver=training/net1/solver.prototxt --gpu=0

• train net2

caffe train --solver=training/net2/vanilla_adam_solver.prototxt --gpu=0

How to testing or predict

cd testing
python test.py ../model/net1/_iter_100000.caffemodel ../training/net1/deploy.prototxt

Please replace correct path of caffe model and deploy file in above command.

How to do a benchmark

cd benchmark
python test.py ../model/net1/_iter_100000.caffemodel ../training/net1/deploy.prototxt

Please replace correct path of caffe model and deploy file in above command.
After do that, you will get mean error of your model.

Reference

VanillaCNN