/FacialLandmark_Caffe

Facial Landmark Detection with Caffe CNN. Face alignment avaiable.

Primary LanguagePython

What's this

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

How to prepare dataset

  • Please download CelebA dataset
  • change input path to your dataset path in train.prototxt file.

How to training

Please run these command in the root of this project. Note: Please change parameter in train.prototxt. such as process num and buffer2memory!

  • set env
export PYTHONPATH=`pwd`/common/:$PYTHONPATH
  • 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

Other DL framework

  • The tensorlfow implementation to see here
  • The mxnet implementation to see here