akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adjust args microphone stream example, util function bugfix [skip ci]

Browse source
This commit is contained in:
dscripka 2023-06-13 21:23:57 -04:00
parent c8833f12b2
commit cdd1e529ab
2 changed files with 29 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

25
examples/detect_from_microphone.py
View file

@ -27,6 +27,20 @@ parser.add_argument(
default=1280,
required=True
)
parser.add_argument(
"--model_path",
help="The path of a specific model to load",
type=str,
default="",
required=False
)
parser.add_argument(
"--inference_framework",
help="The inference framework to use (either 'onnx' or 'tflite'",
type=str,
default='tflite',
required=False
)
args=parser.parse_args()
@ -39,7 +53,12 @@ audio = pyaudio.PyAudio()
mic_stream = audio.open(format=FORMAT, channels=CHANNELS, rate=RATE, input=True, frames_per_buffer=CHUNK)
# Load pre-trained openwakeword models
owwModel = Model()
if args.model_path != "":
owwModel = Model(wakeword_models=[args.model_path], inference_framework=args.inference_framework)
else:
owwModel = Model(inference_framework=args.inference_framework)
n_models = len(owwModel.models.keys())
# Run capture loop continuosly, checking for wakewords
if __name__ == "__main__":
@ -48,7 +67,7 @@ if __name__ == "__main__":
print("#"*100)
print("Listening for wakewords...")
print("#"*100)
print("\n"*13)
print("\n"*(n_models*3))
while True:
# Get audio
@ -73,5 +92,5 @@ if __name__ == "__main__":
"""
# Print results table
print("\033[F"*14)
print("\033[F"*(4*n_models+1))
print(output_string_header, " ", end='\r')