Gardenia is a face alignment project using Convolutional Neural Network. The project illstrate the training and testing process on 300W dataset.
- Download and install [caffe from xuzhenqi] (https://github.com/xuzhenqi/caffe/tree/eye_detection). We use both the c++ and python interfaces.
- Matlab
We need first to prepare the data, suppose the original data root path are data/. Run
scripts/prepare.sh
To crop the testset and augment the trainset, and we will get the data/augment/train.txt, data/train_mean.blob and data/crop/test.txt.
After preparing the data, we can run
scripts/train.sh
to train our model. The prototxt are set done on proto folder.
After training a model, suppose the model path to be $MODEL, run
scripts/inference.sh # You may need to set some options in inference.sh to predict labels
scripts/evalution.sh # report mean err on 300W dataset and CED curve.
The inference.sh will use trainded model to predict shapes, the pretrainded can be downloaded from [here] (http://pan.baidu.com/s/1cBOph8). Feel free to put an issue if you encounter problems.