name
This commit is contained in:
6
demo.py
6
demo.py
@ -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)
|
||||
|
||||
Reference in New Issue
Block a user