-
Notifications
You must be signed in to change notification settings - Fork 0
/
create_imgtxt_16k.py
43 lines (30 loc) · 1.04 KB
/
create_imgtxt_16k.py
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
32
33
34
35
36
37
38
39
40
41
42
43
import os
SCRIPT_DIR = os.getcwd()
TRAINING_FOLDER = 'Input_spectrogram_16k/Training/'
languages = os.listdir(TRAINING_FOLDER)
languages.sort()
spec_dict_train = {}
for l in languages:
spec_dict_train[l] = sorted(os.listdir(TRAINING_FOLDER + l))
txtfile = open('./img_set_16k_train.txt', mode='w')
for l in languages:
os.chdir(TRAINING_FOLDER + str(l))
for i in range(0, len(spec_dict_train[l])):
txtfile.write(str(spec_dict_train[l][i]) + " " + str(l))
txtfile.write("\n")
os.chdir(SCRIPT_DIR)
txtfile.close()
VALIDATION_FOLDER = 'Input_spectrogram_16k/Validation/'
languages = os.listdir(VALIDATION_FOLDER)
languages.sort()
spec_dict_val = {}
for l in languages:
spec_dict_val[l] = sorted(os.listdir(VALIDATION_FOLDER + l))
txtfile = open('./img_set_16k_val.txt', mode='w')
for l in languages:
os.chdir(VALIDATION_FOLDER + str(l))
for i in range(0, len(spec_dict_val[l])):
txtfile.write(str(spec_dict_val[l][i]) + " " + str(l))
txtfile.write("\n")
os.chdir(SCRIPT_DIR)
txtfile.close()