lmd/Pose_to_SMPL_an_230402
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support CMU Mocap

Browse Source
This commit is contained in:
Iridoudou
2021-08-11 12:12:31 +08:00
parent 3bb691ee4d
commit dd425a66d3
13 changed files with 1311 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
fit/tools/save.py
View File

@ -33,17 +33,17 @@ def save_pic(res, smpl_layer, file, logger, dataset_name,target):
savepath=os.path.join(fit_path+"/frame_{}".format(i)),
batch_idx=i,
show=False,
only_joint=False)
display_model(
{'verts': verts.cpu().detach(),
'joints': target.cpu().detach()},
model_faces=smpl_layer.th_faces,
with_joints=True,
kintree_table=smpl_layer.kintree_table,
savepath=os.path.join(gt_path+"/frame_{}".format(i)),
batch_idx=i,
show=False,
only_joint=True)
# display_model(
# {'verts': verts.cpu().detach(),
# 'joints': target.cpu().detach()},
# model_faces=smpl_layer.th_faces,
# with_joints=True,
# kintree_table=smpl_layer.kintree_table,
# savepath=os.path.join(gt_path+"/frame_{}".format(i)),
# batch_idx=i,
# show=False,
# only_joint=True)
logger.info('Pictures saved')