search

abhinanda-punnakkal / BABEL

Code accompanying the BABEL dataset (CVPR 2021).
156 stars 9 forks source link

how to generate joint_pose in create_dataset.py ？ #12

Closed MengHao666 closed 1 year ago

MengHao666 commented 1 year ago

Could you provide the code of generte it ? I am very confused about it. I could only get 52 joints of SMPLH model with human-body-prior. How could you do it?

carjun commented 1 year ago

Hi @MengHao666 ! (1) You might find code in the SMPLX repo useful if you'd like to generate joint positions from the AMASS npz file with joint rotations. Specifically, instantiate the SMPLH body model (class definition is here). Then, call the forward function here with the pose (rotations) as argument. This function should return the joint positions. (2) Yes, there is no 1-1 mapping between all the joints in SMPLH skeleton format and the NTU skeleton format. So to get the locations of certain joints of the NTU skeleton, we actually obtain the corresponding vertices from the full mesh. We obtain the locations in the mesh using the same forward function described in (1).