diff --git a/skel/alignment/utils.py b/skel/alignment/utils.py index b7e0e58..bd70ed3 100644 --- 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