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

Fix copying remote AudioFile target to local

Browse Source
This commit is contained in:
CatalinVoss 2021-01-22 14:34:49 -08:00 committed by Reuben Morais
parent efbd6be727
commit f27908e7e3
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -174,6 +174,10 @@ def convert_audio(src_audio_path, dst_audio_path, file_type=None, audio_format=D
class AudioFile: class AudioFile:
"""
Audio data file wrapper that ensures that the file is loaded with the correct sample rate, channels,
and width, and converts the file on the fly otherwise.
"""
def __init__(self, audio_path, as_path=False, audio_format=DEFAULT_FORMAT): def __init__(self, audio_path, as_path=False, audio_format=DEFAULT_FORMAT):
self.audio_path = audio_path self.audio_path = audio_path
self.audio_format = audio_format self.audio_format = audio_format
@ -199,8 +203,8 @@ class AudioFile:
# If the format isn't right, copy the file to local tmp dir and do the conversion on disk # If the format isn't right, copy the file to local tmp dir and do the conversion on disk
if is_remote_path(self.audio_path): if is_remote_path(self.audio_path):
_, self.tmp_src_file_path = tempfile.mkstemp(suffix='.wav') _, self.tmp_src_file_path = tempfile.mkstemp(suffix='.wav')
copy_remote(self.audio_path, self.tmp_src_file_path) copy_remote(self.audio_path, self.tmp_src_file_path, True)
self.audio_path = self.tmp_file_path self.audio_path = self.tmp_src_file_path
_, self.tmp_file_path = tempfile.mkstemp(suffix='.wav') _, self.tmp_file_path = tempfile.mkstemp(suffix='.wav')
convert_audio(self.audio_path, self.tmp_file_path, file_type='wav', audio_format=self.audio_format) convert_audio(self.audio_path, self.tmp_file_path, file_type='wav', audio_format=self.audio_format)