Handle no translation file

MarilynKeller · Oct 23, 2024 · cb00126 · cb00126
1 parent 827f497
commit cb00126
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skel/alignment/utils.py b/skel/alignment/utils.py
@@ -85,8 +85,8 @@ def load_smpl_seq(smpl_seq_path, gender=None, straighten_hands=False):
     elif 'transl' in data_dict:
         data_fixed['trans'] = data_dict['transl']
     else:
-        raise Exception(f"Could not find trans in {smpl_seq_path}. Available keys: {data_dict.keys()})")
-        # data_fixed['trans'] = np.zeros((poses.shape[0], 3))
+        print(f'WARNING: Could not find translation in {smpl_seq_path}. Setting translation to zeros.')
+        data_fixed['trans'] = np.zeros((poses.shape[0], 3))
 
     # Get betas 
     betas = data_dict['betas'][..., :10] # Keep only the 10 first betas