ledduy610 / accurate-head-pose

Pytorch code for Hybrid Coarse-fine Classification for Head Pose Estimation

accurate-head-pose

We release the code of the Hybrid Coarse-fine Classification for Head Pose Estimation, built on top of the deep-head-pose.

Pretrained model

We provide pretrained model to reproduce the same result shown in the paper.

AFLW2000, password: drmz

AFLW, password: yym5

BIWI, password: 8qpc

Testing

Training and testing lists can be found in /tools, you need download corresonding dataset and update the path.
AFLW2000 dataset, password: xr6e

python test_hopenet.py --gpu 0 --data_dir directory-path-for-dataset --filename_list filename-list --snapshot model --dataset dataset-name 

TODO

Instructions for scripts
Better and better models
Videos and example demo

Cite this work

Haofan Wang, Zhenhua Chen and Yi Zhou "Hybrid coarse-fine classification for head pose estimation." arXiv:1901.06778, 2019. (Download)

Biblatex entry:

        @article{wang2019hybrid,
          title={Hybrid coarse-fine classification for head pose estimation},
          author={Wang, Haofan and Chen, Zhenghua and Zhou, Yi},
          journal={arXiv preprint arXiv:1901.06778},
          year={2019}
        }

Acknowledgement

Our hybrid classification network is plug-and-play on top of the deep-head-pose, but it could be extended to other classification tasks easily. We thank Nataniel Ruiz for releasing deep-head-pose-Pytorch codebase.