search

mkocabas / VIBE

Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"
https://arxiv.org/abs/1912.05656
Other
2.89k stars 549 forks source link

threedpw_utils.py 疑惑 #160

Open zhLawliet opened 3 years ago

zhLawliet commented 3 years ago

threedpw_utils.py

采用默认的threedpw_utils代码处理3dpw的数据时候，采用下面的代码获得vertices

output = smpl(betas=shape, body_pose=pose[:,3:], global_orient=pose[:,:3], transl=trans)#原始代码

发现这个vertices与spin网络获得cam并不一致，原因是这里传入了3dpw的trans参数， 而vibe模型参数的中，网络训练的输出，并没有传入trans参数，代码如下，不知道是不是弄错了， 或者在代码里面哪里做了转换，望指教，谢谢 pred_output = self.smpl( betas=pred_shape, body_pose=pred_rotmat[:, 1:], global_orient=pred_rotmat[:, 0].unsqueeze(1), pose2rot=False )

ghost commented 3 years ago

这代码写得其乱无比。。。 看着真是脑仁疼