Deep Head Pose V2
Welcome to the deep-head-pose-v2 repo.
The work is described in the pdf file in the root of the project.
We also share our trained models that can be used to validate results from the paper:
To perform inference on one of the models, on the AFLW2000 dataset use:
python test.py --data_dir path/to/data --filename_list path/to/filenames --dataset "AFLW2000" --model 'ResNet50' --snapshot path/to/master50.pkl
To train a model from scratch, do
python train_hopenet.py --data_dir path/to/data --filename_list path/to/filenames --num_epochs 35 --model 'ResNet50' ...