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-14 16:24:48 +08:00
parent f9c327d2a2
commit a5f0cf4653
4 changed files with 38 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fit/tools/save.py
View File

@ -6,8 +6,8 @@ import numpy as np
import pickle
sys.path.append(os.getcwd())
from display_utils import display_model
from label import get_label
from display_utils import display_model
def create_dir_not_exist(path):
@ -15,11 +15,11 @@ def create_dir_not_exist(path):
os.mkdir(path)
def save_pic(res, smpl_layer, file, logger, dataset_name,target):
def save_pic(res, smpl_layer, file, logger, dataset_name, target):
_, _, verts, Jtr = res
file_name = re.split('[/.]', file)[-2]
fit_path = "fit/output/{}/picture/fit/{}".format(dataset_name,file_name)
gt_path = "fit/output/{}/picture/gt/{}".format(dataset_name,file_name)
fit_path = "fit/output/{}/picture/fit/{}".format(dataset_name, file_name)
gt_path = "fit/output/{}/picture/gt/{}".format(dataset_name, file_name)
create_dir_not_exist(fit_path)
create_dir_not_exist(gt_path)
logger.info('Saving pictures at {}'.format(fit_path))
@ -53,7 +53,7 @@ def save_params(res, file, logger, dataset_name):
fit_path = "fit/output/{}/".format(dataset_name)
create_dir_not_exist(fit_path)
logger.info('Saving params at {}'.format(fit_path))
label=get_label(file_name, dataset_name)
label = get_label(file_name, dataset_name)
pose_params = (pose_params.cpu().detach()).numpy().tolist()
shape_params = (shape_params.cpu().detach()).numpy().tolist()
Jtr = (Jtr.cpu().detach()).numpy().tolist()
@ -64,5 +64,5 @@ def save_params(res, file, logger, dataset_name):
params["shape_params"] = shape_params
params["Jtr"] = Jtr
with open(os.path.join((fit_path),
"{}_params.pkl".format(file_name)), 'wb') as f:
"{}_params.pkl".format(file_name)), 'wb') as f:
pickle.dump(params, f)