add models and convert speed to ms

2020-02-18 05:55:06 -06:00
parent 04e9ab5ce4
commit bb8e4621f5
5 changed files with 17 additions and 20 deletions
--- a/frigate/edgetpu.py
+++ b/frigate/edgetpu.py
@@ -36,10 +36,10 @@ class ObjectDetector():
        
        if edge_tpu_delegate is None:
            self.interpreter = tflite.Interpreter(
-                model_path=model_file)
+                model_path='/cpu_model.tflite')
        else:
            self.interpreter = tflite.Interpreter(
-                model_path=model_file,
+                model_path='/edgetpu_model.tflite',
                experimental_delegates=[edge_tpu_delegate])
        
        self.interpreter.allocate_tensors()