-
-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathexample_landmarks.py
More file actions
31 lines (24 loc) · 741 Bytes
/
Copy pathexample_landmarks.py
File metadata and controls
31 lines (24 loc) · 741 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import pyVSR
from pyVSR import tcdtimit
from os import path
def main():
dataset_dir = '/run/media/john_tukey/download/datasets/adapt/tcdtimit/'
train, test = tcdtimit.files.request_files(
dataset_dir=dataset_dir,
protocol='speaker_dependent',
remove_sa=True
)
experiment = pyVSR.AVSR(num_threads=4)
file_dict = dict()
output_dir = './run/features/landmarks/'
for file in (train+test):
file_dict[file] = path.join(
output_dir,
path.splitext(file.split('tcdtimit/')[-1])[0] + '.csv')
experiment.extract_save_features(
files=file_dict,
feature_type='landmarks',
extract_opts=None,
)
if __name__ == "__main__":
main()