update

open-mmlab · May 23, 2024 · cc95c90 · cc95c90
1 parent a7e1929
commit cc95c90
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mmhuman3d/data/data_converters/synbody.py b/mmhuman3d/data/data_converters/synbody.py
@@ -309,11 +309,13 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
                             self.smplx_shape[k]))
                 # pdb.set_trace()
                 gender = []
+                track_ids = []
                 for idx, meta_tmp in enumerate(npfile['meta'][valid_id]):
                     gender.append(meta_tmp['gender'])
+                    track_ids.append(track_id * 100 + meta_tmp['track_id'])
 
                 _meta['gender'] += gender
-                _meta['track_id'] += [track_id] * len(gender)
+                _meta['track_id'] += track_ids
                 if len(list(set(gender))) != 1:
                     print(f'{npzf} has more than 1 gender, please check')
                 # pdb.set_trace()