fix

open-mmlab · Oct 31, 2023 · 870dcf7 · 870dcf7
1 parent 0c13b0b
commit 870dcf7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/mmhuman3d/data/data_converters/agora.py b/mmhuman3d/data/data_converters/agora.py
@@ -198,8 +198,12 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
                                       anno_info['smplx_param_path'])
 
             # smplx_param = pickle.load(open(smplx_path, 'rb'))
-            smplx_param_bf = pickle.load(open(
-                os.path.join(dataset_path, anno_info_bf['smplx_param_path']), 'rb'))
+            try:
+                smplx_param_bf = pickle.load(open(
+                    os.path.join(dataset_path, anno_info_bf['smplx_param_path']), 'rb'))
+            except:
+                print(f'{anno_info_bf["smplx_param_path"]}, not found')
+                continue
             smplx_param = smplx_param_bf
             smplx_param['betas_fixed'] = smplx_param_bf['betas'][:, :10]