fix

open-mmlab · Oct 31, 2023 · 060308c · 060308c
1 parent 5829692
commit 060308c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmhuman3d/data/data_converters/agora.py b/mmhuman3d/data/data_converters/agora.py
@@ -124,7 +124,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
         ]:
             bboxs_[bbox_name] = []
         meta_ = {}
-        for meta_key in ['principal_point', 'focal_length', 'gender']:
+        for meta_key in ['principal_point', 'focal_length', 'gender', 'annot_valid']:
             meta_[meta_key] = []
         image_path_ = []
 
@@ -172,7 +172,6 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
         # for aid, anno_info in enumerate(tqdm(anno_param, desc=f'Processing Agora {mode}')):
             image_id = anno_info['image_id']
 
-            pdb.set_trace()
             # anno_info_bf = anno_param_bf[aid]
 
             # print(anno_info['gender'])
@@ -232,6 +231,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
             meta_['gender'].append(anno_info['gender'])
             meta_['principal_point'].append(principal_point)
             meta_['focal_length'].append(focal_length)
+            meta_['annot_valid'].append(anno_info['is_valid'])
 
             # print(smplx_param['betas'].shape)