HoyerChou/DirectMHP

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.

Single HPE Task Pretrained on WiderFace and Finetuning on 300W-LP

• DirectMHP-S --> 300wlp_s_512_e50_finetune_best.pt
• DirectMHP-M --> 300wlp_m_512_e50_finetune_best.pt

Single HPE SixDRepNet Re-trained on AGORA-HPE and CMU-HPE

• AGORA-HPE --> SixDRepNet_AGORA_bs256_e100_epoch_last.pth
• CMU-HPE --> SixDRepNet_CMU_bs256_e100_epoch_last.pth