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

Address pull request feedback from @lissyx

Browse Source
This commit is contained in:
Jordan Olafsen 2019-03-04 23:00:00 +13:00
parent dedf2911da
commit 77b2f8f3ec
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/mic_vad_streaming/mic_vad_streaming.py Normal file → Executable file
View File

@ -11,7 +11,6 @@ from scipy import signal
logging.basicConfig(level=20) logging.basicConfig(level=20)
class Audio(object): class Audio(object):
"""Streams raw audio from microphone. Data is received in a separate thread, and stored in a buffer, to be read from.""" """Streams raw audio from microphone. Data is received in a separate thread, and stored in a buffer, to be read from."""
@ -25,8 +24,7 @@ class Audio(object):
def proxy_callback(in_data, frame_count, time_info, status): def proxy_callback(in_data, frame_count, time_info, status):
callback(in_data) callback(in_data)
return (None, pyaudio.paContinue) return (None, pyaudio.paContinue)
if callback is None: if callback is None: callback = lambda in_data: self.buffer_queue.put(in_data)
callback = lambda in_data: self.buffer_queue.put(in_data)
self.buffer_queue = queue.Queue() self.buffer_queue = queue.Queue()
self.device = device self.device = device
self.input_rate = input_rate self.input_rate = input_rate
@ -117,8 +115,7 @@ class VADAudio(Audio):
Example: (frame, ..., frame, None, frame, ..., frame, None, ...) Example: (frame, ..., frame, None, frame, ..., frame, None, ...)
|---utterence---| |---utterence---| |---utterence---| |---utterence---|
""" """
if frames is None: if frames is None: frames = self.frame_generator()
frames = self.frame_generator()
num_padding_frames = padding_ms // self.frame_duration_ms num_padding_frames = padding_ms // self.frame_duration_ms
ring_buffer = collections.deque(maxlen=num_padding_frames) ring_buffer = collections.deque(maxlen=num_padding_frames)
triggered = False triggered = False
@ -190,7 +187,6 @@ def main(ARGS):
print("Recognized: %s" % text) print("Recognized: %s" % text)
stream_context = model.setupStream() stream_context = model.setupStream()
if __name__ == '__main__': if __name__ == '__main__':
BEAM_WIDTH = 500 BEAM_WIDTH = 500
DEFAULT_SAMPLE_RATE = 16000 DEFAULT_SAMPLE_RATE = 16000
@ -218,7 +214,7 @@ if __name__ == '__main__':
parser.add_argument('-t', '--trie', default='trie', parser.add_argument('-t', '--trie', default='trie',
help="Path to the language model trie file created with native_client/generate_trie. Default: trie") help="Path to the language model trie file created with native_client/generate_trie. Default: trie")
parser.add_argument('-d', '--device', type=int, default=None, parser.add_argument('-d', '--device', type=int, default=None,
help="Device input index (Int) as listed by pyaudio.PyAudio.get_device_info_by_index()") help="Device input index (Int) as listed by pyaudio.PyAudio.get_device_info_by_index(). If not provided, falls back to PyAudio.get_default_device()")
parser.add_argument('-r', '--rate', type=int, default=DEFAULT_SAMPLE_RATE, parser.add_argument('-r', '--rate', type=int, default=DEFAULT_SAMPLE_RATE,
help=f"Input device sample rate. Default: {DEFAULT_SAMPLE_RATE}. Your device may require 44100.") help=f"Input device sample rate. Default: {DEFAULT_SAMPLE_RATE}. Your device may require 44100.")
parser.add_argument('-nf', '--n_features', type=int, default=N_FEATURES, parser.add_argument('-nf', '--n_features', type=int, default=N_FEATURES,