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

Support NTU

Browse Source
This commit is contained in:
Iridoudou
2021-09-03 15:31:39 +08:00
parent 52603950d6
commit fd2ff6616f
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fit/configs/NTU.json
View File

@ -7,12 +7,12 @@
"MAX_EPOCH": 1000,
"WRITE": 10,
"OPTIMIZE_SCALE":0,
"OPTIMIZE_SHAPE":1
"OPTIMIZE_SHAPE":0
},
"USE_GPU": 1,
"DATASET": {
"NAME": "NTU",
"PATH": "../NTU RGB+D/skeleton_npy",
"PATH": "../NTU RGB+D/result",
"TARGET_PATH": "",
"DATA_MAP": [
[

2
fit/tools/main.py
View File

@ -115,7 +115,7 @@ if __name__ == "__main__":
logger.info("avg_loss:{:.4f}".format(meters.avg))
save_params(res, file, logger, args.dataset_name)
save_pic(res,smpl_layer,file,logger,args.dataset_name,target)
# save_pic(res,smpl_layer,file,logger,args.dataset_name,target)
torch.cuda.empty_cache()
logger.info("Fitting finished! Average loss: {:.9f}".format(meters.avg))