From de5de1b082802d6fe51cf4cf6ca969e49e2ea52a Mon Sep 17 00:00:00 2001 From: Wei-Chen-hub <1259566226@qq.com> Date: Wed, 6 Mar 2024 10:56:39 +0800 Subject: [PATCH] update --- mmhuman3d/data/data_converters/synbody_whac.py | 5 ++++- tools/convert_datasets.py | 3 ++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/mmhuman3d/data/data_converters/synbody_whac.py b/mmhuman3d/data/data_converters/synbody_whac.py index 7fb32bea..336bdd1a 100644 --- 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 index 7a86058e..49d21d59 100644 --- 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',