search

deepinsight / insightface

State-of-the-art 2D and 3D Face Analysis Project
https://insightface.ai
22.98k stars 5.37k forks source link

Predict head pose from 2d106det.onnx ? #2601

Open vadik6666 opened 3 months ago

vadik6666 commented 3 months ago

Hi, how can I predict head pose (roll, yaw, pitch) from landmarks predicted by 2d106det.onnx ?

I see in the code that I need 1k3d68.onnx model for that, but that model is 140 Mb while 2d106det.onnx is only 5Mb.

I simply need to predict head pose fast and using small model

nttstar commented 1 month ago

predicting head pose using 2d landmark is not quite accurate.