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

Fix wave file reading helpers

Browse Source
This commit is contained in:
CatalinVoss 2020-11-12 16:40:40 -08:00
parent 783cdad8db
commit 8fe972eb6f
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
training/deepspeech_training/util/audio.py
View File

@ -336,7 +336,8 @@ def read_opus(opus_file):
def write_wav(wav_file, pcm_data, audio_format=DEFAULT_FORMAT):
with wave.open_remote(wav_file, 'wb') as wav_file_writer:
# wav_file is already a file-pointer here
with wave.open(wav_file, 'wb') as wav_file_writer:
wav_file_writer.setframerate(audio_format.rate)
wav_file_writer.setnchannels(audio_format.channels)
wav_file_writer.setsampwidth(audio_format.width)
@ -345,7 +346,7 @@ def write_wav(wav_file, pcm_data, audio_format=DEFAULT_FORMAT):
def read_wav(wav_file):
wav_file.seek(0)
with wave.open_remote(wav_file, 'rb') as wav_file_reader:
with wave.open(wav_file, 'rb') as wav_file_reader:
audio_format = read_audio_format_from_wav_file(wav_file_reader)
pcm_data = wav_file_reader.readframes(wav_file_reader.getnframes())
return audio_format, pcm_data
@ -369,7 +370,7 @@ def write_audio(audio_type, audio_file, pcm_data, audio_format=DEFAULT_FORMAT, b
def read_wav_duration(wav_file):
wav_file.seek(0)
with wave.open_remote(wav_file, 'rb') as wav_file_reader:
with wave.open(wav_file, 'rb') as wav_file_reader:
return wav_file_reader.getnframes() / wav_file_reader.getframerate()