Update README.md

README.md

@@ -1,3 +1,19 @@
 ---
 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](./cmu_s_1280_e200_t40_lw010_best.pt)
+* DirectMHP-M --> [cmu_m_1280_e200_t40_lw010_best.pt](./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](./agora_s_1280_e300_t40_lw010_best.pt)
+* DirectMHP-M --> [agora_m_1280_e300_t40_lw010_best.pt](./agora_m_1280_e300_t40_lw010_best.pt)