diff --git a/parakeet/datasets/ljspeech.py b/parakeet/datasets/ljspeech.py index 62209e9219a50855fbd94f7deab96537407c4d18..3ab8ac92edd06b5411033ea9407545aa6ee6dd75 100644 --- a/parakeet/datasets/ljspeech.py +++ b/parakeet/datasets/ljspeech.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from pathlib import Path +import os import numpy as np import pandas as pd import librosa @@ -27,13 +27,11 @@ from ..data.batch import TextIDBatcher, SpecBatcher class LJSpeech(DatasetMixin): def __init__(self, root): super(LJSpeech, self).__init__() - assert isinstance(root, ( - str, Path)), "root should be a string or Path object" - self.root = root if isinstance(root, Path) else Path(root) + self.root = root self.metadata = self._prepare_metadata() def _prepare_metadata(self): - csv_path = self.root.joinpath("metadata.csv") + csv_path = os.path.join(self.root, "metadata.csv") metadata = pd.read_csv( csv_path, sep="|", @@ -51,7 +49,7 @@ class LJSpeech(DatasetMixin): """ fname, raw_text, normalized_text = metadatum - wav_path = self.root.joinpath("wavs", fname + ".wav") + wav_path = os.path.join(self.root, "wavs", fname + ".wav") # load -> trim -> preemphasis -> stft -> magnitude -> mel_scale -> logscale -> normalize wav, sample_rate = librosa.load( diff --git a/parakeet/models/waveflow/data.py b/parakeet/models/waveflow/data.py index 0c1e914271214de93de514098872183b94843375..83438f79d6618e21483fa583f86946fc5cf9e023 100644 --- a/parakeet/models/waveflow/data.py +++ b/parakeet/models/waveflow/data.py @@ -12,6 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +import os import random import librosa @@ -32,7 +33,7 @@ class Dataset(ljspeech.LJSpeech): def _get_example(self, metadatum): fname, _, _ = metadatum - wav_path = self.root.joinpath("wavs", fname + ".wav") + wav_path = os.path.join(self.root, "wavs", fname + ".wav") loaded_sr, audio = read(wav_path) assert loaded_sr == self.config.sample_rate