fulyvv/EasyMocap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix read smpl bug

Browse Source
This commit is contained in:
Qing Shuai 2022-08-23 21:05:35 +08:00
parent 0175f07290
commit de973d4047
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/postprocess/convert2bvh.py
View File

@ -152,6 +152,8 @@ def read_smpl(outname):
assert os.path.exists(outname), outname assert os.path.exists(outname), outname
datas = read_json(outname) datas = read_json(outname)
outputs = [] outputs = []
if isinstace(datas, dict):
datas = datas['annots']
for data in datas: for data in datas:
for key in ['Rh', 'Th', 'poses', 'shapes']: for key in ['Rh', 'Th', 'poses', 'shapes']:
data[key] = np.array(data[key]) data[key] = np.array(data[key])