fix read smpl bug

2022-08-23 21:05:35 +08:00 · 2022-08-23 21:05:35 +08:00 · de973d4047
commit de973d4047
parent 0175f07290
1 changed files with 3 additions and 1 deletions
--- a/scripts/postprocess/convert2bvh.py
+++ b/scripts/postprocess/convert2bvh.py
@ -152,6 +152,8 @@ def read_smpl(outname):
    assert os.path.exists(outname), outname
    datas = read_json(outname)
    outputs = []
+    if isinstace(datas, dict):
+        datas = datas['annots']
    for data in datas:
        for key in ['Rh', 'Th', 'poses', 'shapes']:
            data[key] = np.array(data[key])