vixy/headmic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable Edge TPU for YAMNet sound classification

Browse Source 
Prefer yamnet_edgetpu.tflite when available, fall back to CPU model.
~50-100ms → ~2-3ms inference per classification.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alex
2026-04-11 16:47:27 -05:00
parent f41b852b5d
commit 73b6793c02
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
headmic.py
View File

@@ -465,15 +465,19 @@ async def startup():
# --- Sound classifier (optional) --- # --- Sound classifier (optional) ---
model_dir = Path(__file__).parent / "models" model_dir = Path(__file__).parent / "models"
edgetpu_model_path = model_dir / "yamnet_edgetpu.tflite"
model_path = model_dir / "yamnet.tflite" model_path = model_dir / "yamnet.tflite"
class_map_path = model_dir / "yamnet_class_map.csv" class_map_path = model_dir / "yamnet_class_map.csv"
if model_path.exists() and class_map_path.exists(): # Prefer Edge TPU model if available
use_edgetpu = edgetpu_model_path.exists()
active_model = edgetpu_model_path if use_edgetpu else model_path
if active_model.exists() and class_map_path.exists():
try: try:
from sound_id import SoundClassifier from sound_id import SoundClassifier
sound_classifier = SoundClassifier(str(model_path), str(class_map_path)) sound_classifier = SoundClassifier(str(active_model), str(class_map_path), use_edgetpu=use_edgetpu)
sound_ring_buffer = collections.deque(maxlen=31) sound_ring_buffer = collections.deque(maxlen=31)
state.sound_classification_enabled = True state.sound_classification_enabled = True
logger.info("Sound classification enabled (YAMNet)") logger.info("Sound classification enabled (YAMNet %s)", "Edge TPU" if use_edgetpu else "CPU")
sc_thread = threading.Thread(target=sound_classifier_loop, daemon=True) sc_thread = threading.Thread(target=sound_classifier_loop, daemon=True)
sc_thread.start() sc_thread.start()