From cc95c906cf4cbf889e6958040aad149e9a6b95d4 Mon Sep 17 00:00:00 2001 From: Wei-Chen-hub <1259566226@qq.com> Date: Thu, 23 May 2024 14:43:44 +0800 Subject: [PATCH] update --- mmhuman3d/data/data_converters/synbody.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mmhuman3d/data/data_converters/synbody.py b/mmhuman3d/data/data_converters/synbody.py index 5edb4c7b..268606a9 100644 --- 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()