experiments
/
STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #3179 from tilmankamp/fix-for-librosa 

Fixes #3178 - Librosa requires 1-dimensional array for mono samples
This commit is contained in:
Tilman Kamp 2020-07-24 14:32:17 +02:00 committed by GitHub
parent 9e023660ef ecbdf46940
commit 15a624134a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
training/deepspeech_training/util/augmentations.py
View File

@ -349,8 +349,13 @@ class Resample(SampleAugmentation):
audio = sample.audio audio = sample.audio
orig_len = len(audio) orig_len = len(audio)
audio = np.swapaxes(audio, 0, 1) audio = np.swapaxes(audio, 0, 1)
audio = resample(audio, sample.audio_format.rate, rate) if audio.shape[0] < 2:
audio = resample(audio, rate, sample.audio_format.rate) # since v0.8 librosa enforces a shape of (samples,) instead of (channels, samples) for mono samples
resampled = resample(audio[0], sample.audio_format.rate, rate)
audio[0] = resample(resampled, rate, sample.audio_format.rate)[:orig_len]
else:
audio = resample(audio, sample.audio_format.rate, rate)
audio = resample(audio, rate, sample.audio_format.rate)
audio = np.swapaxes(audio, 0, 1)[0:orig_len] audio = np.swapaxes(audio, 0, 1)[0:orig_len]
sample.audio = audio sample.audio = audio