From 46c002038b172e7e2afc32ef2406a5a922b71ada Mon Sep 17 00:00:00 2001 From: Blake Blackshear Date: Fri, 15 Jan 2021 22:43:42 -0600 Subject: [PATCH] don't fallback to the CPU fixes #381 --- frigate/edgetpu.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/frigate/edgetpu.py b/frigate/edgetpu.py index 79ba626f6..d65ce523b 100644 --- a/frigate/edgetpu.py +++ b/frigate/edgetpu.py @@ -62,16 +62,15 @@ class LocalObjectDetector(ObjectDetector): logger.info(f"Attempting to load TPU as {device_config['device']}") edge_tpu_delegate = load_delegate('libedgetpu.so.1.0', device_config) logger.info("TPU found") + self.interpreter = tflite.Interpreter( + model_path='/edgetpu_model.tflite', + experimental_delegates=[edge_tpu_delegate]) except ValueError: - logger.info("No EdgeTPU detected. Falling back to CPU.") - - if edge_tpu_delegate is None: - self.interpreter = tflite.Interpreter( - model_path='/cpu_model.tflite', num_threads=num_threads) + logger.info("No EdgeTPU detected.") + raise else: self.interpreter = tflite.Interpreter( - model_path='/edgetpu_model.tflite', - experimental_delegates=[edge_tpu_delegate]) + model_path='/cpu_model.tflite', num_threads=num_threads) self.interpreter.allocate_tensors()