bugfix

open-mmlab · Mar 1, 2024 · 0500781 · 0500781
1 parent cb5451a
commit 0500781
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmhuman3d/data/data_converters/synbody_whac.py b/mmhuman3d/data/data_converters/synbody_whac.py
@@ -178,11 +178,11 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
 
         slice_seq_dict = {}
 
-        if mode == 'AMASS_tracking':
+        if 'AMASS_tracking' in mode:
             for slid in range(slices):
                 seqs = seqs_targeted[slid * len(seqs_targeted) // slices: (slid + 1) * len(seqs_targeted) // slices]
                 slice_seq_dict[slid] = seqs
-        elif mode == 'DuetDance':
+        elif 'DuetDance' in mode:
             seq_ns = list(set([os.path.dirname(os.path.dirname(seq)) for seq in seqs_targeted]))
             seq_ns = [os.path.basename(seq_n) for seq_n in seq_ns]
             random.shuffle(seq_ns)