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

Make sure we properly unpack samples when changing audio types

Browse Source
This commit is contained in:
CatalinVoss 2020-11-17 14:44:26 -08:00
parent 9aaa0e406b
commit 24e9e6777c
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -118,15 +118,16 @@ class Sample:
self.audio_type = new_audio_type
def _change_audio_type(sample_and_audio_type):
sample, audio_type, bitrate = sample_and_audio_type
def _unpack_and_change_audio_type(sample_and_audio_type):
packed_sample, audio_type, bitrate = sample_and_audio_type
sample = packed_sample.unpack()
sample.change_audio_type(audio_type, bitrate=bitrate)
return sample
def change_audio_types(samples, audio_type=AUDIO_TYPE_PCM, bitrate=None, processes=None, process_ahead=None):
def change_audio_types(packed_samples, audio_type=AUDIO_TYPE_PCM, bitrate=None, processes=None, process_ahead=None):
with LimitingPool(processes=processes, process_ahead=process_ahead) as pool:
yield from pool.imap(_change_audio_type, map(lambda s: (s, audio_type, bitrate), samples))
yield from pool.imap(_unpack_and_change_audio_type, map(lambda s: (s, audio_type, bitrate), packed_samples))
def get_audio_type_from_extension(ext):