Skip to content

Facial Landmark Detection with Caffe CNN. Face alignment avaiable.

Notifications You must be signed in to change notification settings

BobLiu20/FacialLandmark_Caffe

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Facial Landmark Detection with Caffe CNN. Face alignment avaiable.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages