update

open-mmlab · Mar 1, 2024 · cb5451a · cb5451a
1 parent a7a3461
commit cb5451a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/mmhuman3d/data/data_converters/synbody_whac.py b/mmhuman3d/data/data_converters/synbody_whac.py
@@ -30,7 +30,8 @@
 @DATA_CONVERTERS.register_module()
 class SynbodyWhacConverter(BaseModeConverter):
     """Synbody dataset."""
-    ACCEPTED_MODES = ['AMASS_tracking', 'DuetDance']
+    ACCEPTED_MODES = ['AMASS_tracking-20240221', 'AMASS_tracking-20240229', 
+                      'DuetDance-20240218']
 
     def __init__(self, modes: List = []) -> None:
 
@@ -131,7 +132,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
 
         # get target sequences
         seqs_targeted = glob.glob(os.path.join(dataset_path, 'Synbody_whac', 
-                                               f'{mode}*', '*', 
+                                               f'{mode}', '*', 
                                                '*_*', 'smplx_adjusted', '*.npz'))
 
         # bulid smplx model

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