initial commit

fit/tools/save.py

@@ -0,0 +1,38 @@
+import sys
+import os
+import re
+
+sys.path.append(os.getcwd())
+from display_utils import display_model
+
+def create_dir_not_exist(path):
+    if not os.path.exists(path):
+        os.mkdir(path)
+        
+def save_pic(target, res, smpl_layer, file):
+    pose_params, shape_params, verts, Jtr = res
+    name=re.split('[/.]',file)[-2]
+    gt_path="fit/output/HumanAct12/picture/gt/{}".format(name)
+    fit_path="fit/output/HumanAct12/picture/fit/{}".format(name)
+    create_dir_not_exist(gt_path)
+    create_dir_not_exist(fit_path)
+    for i in range(target.shape[0]):
+        display_model(
+            {'verts': verts.cpu().detach(),
+             'joints': target.cpu().detach()},
+            model_faces=smpl_layer.th_faces,
+            with_joints=True,
+            kintree_table=smpl_layer.kintree_table,
+            savepath=os.path.join(gt_path+"/frame_{}".format(i)),
+            batch_idx=i,
+            show=False,
+            only_joint=True)
+        display_model(
+            {'verts': verts.cpu().detach(),
+             'joints': Jtr.cpu().detach()},
+            model_faces=smpl_layer.th_faces,
+            with_joints=True,
+            kintree_table=smpl_layer.kintree_table,
+            savepath=os.path.join(fit_path+"/frame_{}".format(i)),
+            batch_idx=i,
+            show=False)