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

add file counter

Browse Source
This commit is contained in:
Iridoudou
2021-08-10 10:10:53 +08:00
parent 338cbee387
commit 75607bddec
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fit/tools/main.py
View File

@ -90,9 +90,11 @@ if __name__ == "__main__":
gender=cfg.MODEL.GENDER, gender=cfg.MODEL.GENDER,
model_root='smplpytorch/native/models') model_root='smplpytorch/native/models')
file_num = 0
for root,dirs,files in os.walk(cfg.DATASET.PATH): for root,dirs,files in os.walk(cfg.DATASET.PATH):
for file in files: for file in files:
logger.info('Processing file: {}'.format(file)) file_num += 1
logger.info('Processing file: {} [{} / {}]'.format(file,file_num,len(files)))
target = torch.from_numpy(transform(args.dataset_name, target = torch.from_numpy(transform(args.dataset_name,
load(args.dataset_name, load(args.dataset_name,
os.path.join(root,file)))).float() os.path.join(root,file)))).float()