commit 936bcc90ed3b2e0be61bf1eef48f47669c4863d0
author Scott Main <smain@google.com> Mon Apr 05 22:07:57 2021 +0100
committer Scott Main <smain@google.com> Tue Apr 06 00:34:15 2021 +0100
tree 3363aa0583e13a6855dc86d2c856149075430161
parent e250fc9cf26c9989329de5399765f03738dc9dc2

Add argument to specify mic device

I needed this to use demo with Logitech cam/mic.

Change-Id: I7aa93753589b22011f4e67e328dd3e8fe217343d

voice.py

diff --git a/voice.py b/voice.py
index 316b1d8..f789a4e 100644
--- a/voice.py
+++ b/voice.py
@@ -231,9 +231,9 @@
         prev_detection = detection
 
 class AudioClassifier:
-  def __init__(self, model_file, labels_file):
+  def __init__(self, model_file, labels_file, audio_device_index=0):
     self._thread = threading.Thread(target=classify_audio,
-      args=(model_file, labels_file, self._callback), daemon=True)
+      args=(model_file, labels_file, self._callback, audio_device_index), daemon=True)
     self._queue = queue.Queue()
     self._thread.start()
 

voice_example.py

diff --git a/voice_example.py b/voice_example.py
index f765d59..0c489b9 100644
--- a/voice_example.py
+++ b/voice_example.py
@@ -25,9 +25,10 @@
                        labels_file=voice.VOICE_LABELS,
                        callback=callback)
 
-def run_audio_classifier():    
+def run_audio_classifier():
   c = voice.AudioClassifier(model_file=voice.VOICE_MODEL,
-                            labels_file=voice.VOICE_LABELS)
+                            labels_file=voice.VOICE_LABELS,
+                            audio_device_index=2)
   while True:
     label, score = c.next()
     print(label, score)