Head Pose Estimator on Caffe
The HeadPose Estimator predicts face landmarks (68 points) and head pose (3d pose, yaw, roll, and pitch).
caffe
dlib18.17
(cd your dlib folder
cd python_example
./compile_dlib_python_module.bat
add dlib.so to the PYTHONPATH)
opencv
numpy, pandas
Unzip dlib-18.17.zip
Creat folder "img" to store your testing img
Creat folder "result" to store your visualized results
Creat folder "pose" to store your estimated poses (yaw angles) for each img
Pycharm run dlib-18.17/python_examples/landmarkPredict.py (modify the caffe_root path and testList.txt if needed. testList.txt is a file containing the path of the testing images.)
Modify train.prototxt and train_solver.prototxt files, train the model using your own data.