-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathdata_script.py
80 lines (68 loc) · 3.71 KB
/
data_script.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import os
import pandas as pd
class_ids = {
'real': 0,
'fake': 1
}
# -----------------------------------------------------------------------------------------
# ljspeech
# -----------------------------------------------------------------------------------------
real_entries = os.listdir('/Users/jakegus/Downloads/LJSpeech-1.1/wavs')
real_list = []
for sound in real_entries:
real_data = {'file path': '/Users/jakegus/Downloads/LJSpeech-1.1/wavs/' + sound, 'class': class_ids['real'] }
real_list.append(real_data)
# -----------------------------------------------------------------------------------------
# ljspeech_melgan
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_melgan')
train_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_melgan/' + sound, 'class': class_ids['fake'] }
train_list.append(fake_data)
# -----------------------------------------------------------------------------------------
# ljspeech_waveglow
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_waveglow')
fake_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_waveglow/' + sound, 'class': class_ids['fake'] }
fake_list.append(fake_data)
# -----------------------------------------------------------------------------------------
# ljspeech_full_band_melgan
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_full_band_melgan')
fake_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_full_band_melgan/' + sound, 'class': class_ids['fake'] }
fake_list.append(fake_data)
# -----------------------------------------------------------------------------------------
# ljspeech_hifiGAN
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_hifiGAN')
fake_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_hifiGAN/' + sound, 'class': class_ids['fake'] }
fake_list.append(fake_data)
# -----------------------------------------------------------------------------------------
# ljspeech_multi_band_melgan
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_multi_band_melgan')
fake_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_multi_band_melgan/' + sound, 'class': class_ids['fake'] }
fake_list.append(fake_data)
# -----------------------------------------------------------------------------------------
# ljspeech_parallel_wavegan
# -----------------------------------------------------------------------------------------
fake_entries = os.listdir('/Users/jakegus/Downloads/generated_audio/ljspeech_parallel_wavegan')
fake_list = []
for sound in fake_entries:
fake_data = {'file path': '/Users/jakegus/Downloads/generated_audio/ljspeech_parallel_wavegan/' + sound, 'class': class_ids['fake'] }
fake_list.append(fake_data)
train_real = real_list[0:6549]
test_real = real_list[6550:]
train_data = train_real + train_list
train_df = pd.DataFrame(data = train_data)
test_data = test_real + fake_list
test_df = pd.DataFrame(data = test_data)