From 3f789187159e5bd385f49f26b8643e801d5838a5 Mon Sep 17 00:00:00 2001 From: ForceCore Date: Sun, 31 May 2020 19:51:16 +0900 Subject: [PATCH] Fix phoneme cache file name aliasing problem When the wav file has multiple dots in the file name, _load_or_generate_phoneme_sequence would only use only the first segment of the file name and cause overwrite of *_phoneme.npy --- datasets/TTSDataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/TTSDataset.py b/datasets/TTSDataset.py index 0d884c00..7fe966d7 100644 --- a/datasets/TTSDataset.py +++ b/datasets/TTSDataset.py @@ -92,7 +92,7 @@ class MyDataset(Dataset): return phonemes def _load_or_generate_phoneme_sequence(self, wav_file, text): - file_name = os.path.basename(wav_file).split('.')[0] + file_name = os.path.splitext(os.path.basename(wav_file))[0] cache_path = os.path.join(self.phoneme_cache_path, file_name + '_phoneme.npy') try: