metadata
license: mit
Pretrained models of our method DirectMHP
Title: DirectMHP: Direct 2D Multi-Person Head Pose Estimation with Full-range Angles
Paper link: https://arxiv.org/abs/2302.01110
Code link: https://github.com/hnuzhy/DirectMHP
Mulit-Person Head Pose Estimation Task (trained on CMU-HPE)
- DirectMHP-S --> cmu_s_1280_e200_t40_lw010_best.pt
- DirectMHP-M --> cmu_m_1280_e200_t40_lw010_best.pt
Mulit-Person Head Pose Estimation Task (trained on AGORA-HPE)
- DirectMHP-S --> agora_s_1280_e300_t40_lw010_best.pt
- DirectMHP-M --> agora_m_1280_e300_t40_lw010_best.pt
Single HPE datasets with YOLOv5+COCO format
- Resorted images used in our DirectMHP: 300W-LP.zip, AFLW2000.zip and BIWI_test.zip.
- Resorted corresponding json files: train_300W_LP.json, val_AFLW2000.json and BIWI_test.json.