Ascend-Research / HeadPoseEstimation-WHENet

BSD 3-Clause "New" or "Revised" License
468 stars 76 forks source link

how can i get the 3D position of head (tx,ty,tz) #22

Open eugeneYz opened 2 years ago

eugeneYz commented 2 years ago

Hi , hope for your answer.

vignif commented 2 years ago

Hi! I was also looking for a tool with the full pose information of a human head. Meaning a 6D vector with the position (tx, ty, tz) and orientation (r, p y) with respect to the camera. However, the model in this repo WHENet.h5 is designed to return only yaw pitch and roll, so with the current arch it is not possible to retrieve the position of the head. Here's a snapshot of the command print(model.model.summary())

image

As you see the outputs are just the angles.

Have you found some valid alternatives in the meanwhile?