mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pvd

Browse Source
This commit is contained in:
James Betker 2022-03-05 23:05:29 -07:00
parent d1dc8dbb35
commit 1a05712764
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
codes/data/audio/paired_voice_audio_dataset.py
View File

@ -17,9 +17,17 @@ from utils.util import opt_get
def load_tsv(filename): def load_tsv(filename):
with open(filename, encoding='utf-8') as f: with open(filename, encoding='utf-8') as f:
components = [line.strip().split('\t') for line in f] filepaths_and_text = []
base = os.path.dirname(filename) base = os.path.dirname(filename)
filepaths_and_text = [[os.path.join(base, f'{component[1]}'), component[0]] for component in components] bad_lines = 0
for line in f:
components = line.strip().split('\t')
if len(components) < 3:
bad_lines += 1
if bad_lines > 10:
print(f'{filename} contains 10+ bad entries. Failing. Sample last entry: {line}')
raise ValueError
filepaths_and_text.append([os.path.join(base, f'{components[1]}'), components[0]])
return filepaths_and_text return filepaths_and_text