mirror of https://github.com/coqui-ai/TTS.git
add support fot VCTK and BRSpeech dataset
This commit is contained in:
parent
c4828b2b9e
commit
70c665b9c4
|
@ -229,14 +229,14 @@ def vctk(root_path, meta_files=None, wavs_path='wav48'):
|
|||
items = []
|
||||
meta_files = glob(f"{os.path.join(root_path,'txt')}/**/*.txt", recursive=True)
|
||||
for meta_file in meta_files:
|
||||
_, speaker_id, txt_file = os.path.relpath(meta_file, root_path).split(os.sep)
|
||||
txt, speaker_id, txt_file = os.path.relpath(meta_file,root_path).split(os.sep)
|
||||
file_id = txt_file.split('.')[0]
|
||||
if isinstance(test_speakers, list): # if is list ignore this speakers ids
|
||||
if speaker_id in test_speakers:
|
||||
continue
|
||||
with open(meta_file) as file_text:
|
||||
text = file_text.readlines()[0]
|
||||
wav_file = os.path.join(root_path, wavs_path, speaker_id, file_id+'.wav')
|
||||
wav_file = os.path.join(root_path, wavs_path, speaker_id,file_id+'.wav')
|
||||
items.append([text, wav_file, speaker_id])
|
||||
|
||||
|
||||
return items
|
Loading…
Reference in New Issue