diff options
| author | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-01-15 22:18:02 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp> | 2018-01-15 22:18:02 +0900 |
| commit | 83608c12e7bb28df1966cbe5b9d86a8e23175044 (patch) | |
| tree | 6fc24caaa01d447bf9819bf6c45b3e2d33685579 /become_yukarin/dataset/dataset.py | |
| parent | c0f3eacabde5d41992a5ae1d8d8f0f170f6b155e (diff) | |
超解像可能に
Diffstat (limited to 'become_yukarin/dataset/dataset.py')
| -rw-r--r-- | become_yukarin/dataset/dataset.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/become_yukarin/dataset/dataset.py b/become_yukarin/dataset/dataset.py index 38cf749..ab05568 100644 --- a/become_yukarin/dataset/dataset.py +++ b/become_yukarin/dataset/dataset.py @@ -75,7 +75,7 @@ class WaveFileLoadProcess(BaseDataProcess): self._pad_second = pad_second self._dtype = dtype - def __call__(self, data: str, test): + def __call__(self, data: str, test=None): wave = librosa.core.load(data, sr=self._sample_rate, dtype=self._dtype)[0] if self._top_db is not None: wave = librosa.effects.remix(wave, intervals=librosa.effects.split(wave, top_db=self._top_db)) @@ -92,7 +92,7 @@ class AcousticFeatureProcess(BaseDataProcess): self._alpha = alpha self._dtype = dtype - def __call__(self, data: Wave, test): + def __call__(self, data: Wave, test=None): x = data.wave.astype(numpy.float64) fs = data.sampling_rate |