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

update code

Browse Source
This commit is contained in:
Iridoudou
2021-08-07 21:30:46 +08:00
parent 2b3e65e2a2
commit e68d723056
3 changed files with 5 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
fit/tools/map.py
View File

@ -1,14 +0,0 @@
import numpy as np
def mapping(Jtr,cfg):
name=cfg.DATASET.NAME
if not name=='HumanAct12':
mapped_joint=cfg.DATASET.DATA_MAP.UTD_MHAD
Jtr_mapped=np.zeros([Jtr.shape[0],len(mapped_joint),Jtr.shape[2]])
for i in range(Jtr.shape[0]):
for j in range(len(mapped_joint)):
for k in range(Jtr.shape[2]):
Jtr_mapped[i][j][k]=Jtr[i][mapped_joint[j]][k]
return Jtr_mapped
return Jtr

12
fit/tools/save.py
View File

@ -41,14 +41,10 @@ def save_params(res, file, logger, dataset_name):
fit_path = "fit/output/{}/params/".format(dataset_name)
create_dir_not_exist(fit_path)
logger.info('Saving params at {}'.format(fit_path))
pose_params = pose_params.cpu().detach()
pose_params = pose_params.numpy().tolist()
shape_params = shape_params.cpu().detach()
shape_params = shape_params.numpy().tolist()
Jtr = Jtr.cpu().detach()
Jtr = Jtr.numpy().tolist()
verts = verts.cpu().detach()
verts = verts.numpy().tolist()
pose_params = (pose_params.cpu().detach()).numpy().tolist()
shape_params = (shape_params.cpu().detach()).numpy().tolist()
Jtr = (Jtr.cpu().detach()).numpy().tolist()
verts = (verts.cpu().detach()).numpy().tolist()
params = {}
params["pose_params"] = pose_params
params["shape_params"] = shape_params