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, "MAX_EPOCH": 1000,
"WRITE": 10, "WRITE": 10,
"OPTIMIZE_SCALE":0, "OPTIMIZE_SCALE":0,
"OPTIMIZE_SHAPE":1 "OPTIMIZE_SHAPE":0
}, },
"USE_GPU": 1, "USE_GPU": 1,
"DATASET": { "DATASET": {
"NAME": "NTU", "NAME": "NTU",
"PATH": "../NTU RGB+D/skeleton_npy", "PATH": "../NTU RGB+D/result",
"TARGET_PATH": "", "TARGET_PATH": "",
"DATA_MAP": [ "DATA_MAP": [
[ [

2
fit/tools/main.py
View File

@ -115,7 +115,7 @@ if __name__ == "__main__":
logger.info("avg_loss:{:.4f}".format(meters.avg)) logger.info("avg_loss:{:.4f}".format(meters.avg))
save_params(res, file, logger, args.dataset_name) 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() torch.cuda.empty_cache()
logger.info("Fitting finished! Average loss: {:.9f}".format(meters.avg)) logger.info("Fitting finished! Average loss: {:.9f}".format(meters.avg))