This commit is contained in:
lmd
2025-07-25 15:05:31 +08:00
parent 24e1e31234
commit 65769e5eb6
18 changed files with 3642 additions and 67 deletions

View File

@ -14,10 +14,10 @@ if __name__ == '__main__':
smpl_layer = SMPL_Layer(
center_idx=0,
gender='male',
model_root='smplpytorch/native/models')
model_root='/home/lmd/Code/Pose_to_SMPL_an_230402/smplpytorch/native/models')
# Generate random pose and shape parameters
pose_params = torch.rand(batch_size, 72) * 0.01
pose_params = (torch.rand(batch_size, 72)*2-1) * np.pi
shape_params = torch.rand(batch_size, 10) * 0.03
# GPU mode
@ -36,5 +36,5 @@ if __name__ == '__main__':
model_faces=smpl_layer.th_faces,
with_joints=True,
kintree_table=smpl_layer.kintree_table,
savepath='image.png',
savepath='image1.png',
show=True)