Skip to content

Commit 611da4f

Browse files
committed
Added data parameters script
1 parent 777a499 commit 611da4f

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

data_processing/data_params.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
"""
2+
Shared argument parser for data_params.py and create_vector.py.
3+
4+
By default, we assume that the dataset is found in the <repo>/dataset/raw/ folder,
5+
and the preprocessed datasets will be created in the <repo>/dataset/processed/ folder,
6+
but these paths can be changed with the parameters below.
7+
"""
8+
import argparse
9+
10+
parser = argparse.ArgumentParser(
11+
description="""Parameters for data processing for the paper `Gesticulator:
12+
A framework for semantically-aware speech-driven gesture generation""")
13+
14+
# Folders params
15+
parser.add_argument('--raw_data_dir', '-data_raw', default="../../dataset/raw/",
16+
help='Path to the folder with the raw dataset')
17+
parser.add_argument('--proc_data_dir', '-data_proc', default="../../dataset/processed/",
18+
help='Path to the folder with the processed dataset')
19+
20+
# Sequence processing
21+
parser.add_argument('--seq_len', '-seq_l', default=40,
22+
help='Length of the sequences during training (used only to avoid vanishing gradients)')
23+
parser.add_argument('--n_context', '-n_cont', default=60, type=int,
24+
help='Length of a total past and future context for speech to be used to generate gestures')
25+
26+
# Features
27+
parser.add_argument('--feature_type', '-feat', default="MFCC",
28+
help='''Describes the type of the input features
29+
(can be \'Spectro\', \'MFCC\', \'Pros\', \'MFCC+Pros\' or \'Spectro+Pos\')''')
30+
parser.add_argument('--n_input', '-n_in', default=26, type=int,
31+
help='Number of inputs to the model')
32+
parser.add_argument('--n_output', '-n_out', default=45, type=int,
33+
help='number of outputs of the model')

0 commit comments

Comments
 (0)