update

open-mmlab · Mar 6, 2024 · de5de1b · de5de1b
1 parent f8f5ec6
commit de5de1b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mmhuman3d/data/data_converters/synbody_whac.py b/mmhuman3d/data/data_converters/synbody_whac.py
@@ -31,7 +31,8 @@
 class SynbodyWhacConverter(BaseModeConverter):
     """Synbody dataset."""
     ACCEPTED_MODES = ['AMASS_tracking-20240221', 'AMASS_tracking-20240229', 
-                      'AMASS_tracking-20240301', 'DuetDance-20240218']
+                      'AMASS_tracking-20240301', 'DuetDance-20240218',
+                      'DLP-20240228']
 
     def __init__(self, modes: List = []) -> None:
 
@@ -196,6 +197,8 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
             slice_seq_dict['test'] = []
             for seq_n in seq_ns[int(len(seq_ns) * 0.8):]:
                 slice_seq_dict['test'] += [seq for seq in seqs_targeted if seq_n in seq]       
+        elif 'DLP' in mode:
+            slice_seq_dict[0] = seqs_targeted
 
         for slid, batch in enumerate(slice_seq_dict.keys()):
 

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