update

open-mmlab · Mar 1, 2024 · f901c7a · f901c7a
1 parent 0500781
commit f901c7a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/mmhuman3d/data/data_converters/synbody_whac.py b/mmhuman3d/data/data_converters/synbody_whac.py
@@ -31,7 +31,7 @@
 class SynbodyWhacConverter(BaseModeConverter):
     """Synbody dataset."""
     ACCEPTED_MODES = ['AMASS_tracking-20240221', 'AMASS_tracking-20240229', 
-                      'DuetDance-20240218']
+                      'AMASS_tracking-20240301', 'DuetDance-20240218']
 
     def __init__(self, modes: List = []) -> None:
 
@@ -179,9 +179,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
         slice_seq_dict = {}
 
         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
+            slice_seq_dict[0] = seqs_targeted
         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]

diff --git a/tools/convert_datasets.py b/tools/convert_datasets.py
@@ -118,8 +118,8 @@
     synbody_whac=dict(
         type='SynbodyWhacConverter',  # synthetic
         prefix='synbody',
-        modes=['AMASS_tracking-20240221', 'AMASS_tracking-20240229', 
-                      'DuetDance-20240218']),
+        modes=['AMASS_tracking-20240221', 'AMASS_tracking-20240229',  
+               'AMASS_tracking-20240301', 'DuetDance-20240218']),
     ubody=dict(
         type='UbodyConverter',  # real, has some single
         prefix='ubody',