Browse Source

Added support for PCIe TPU, as well as USB
Also added message showing which found

Carl Elkins 4 years ago
parent
commit
a8556a729b
1 changed files with 8 additions and 3 deletions
  1. 8 3
      frigate/edgetpu.py

+ 8 - 3
frigate/edgetpu.py

@@ -31,9 +31,14 @@ class ObjectDetector():
     def __init__(self):
         edge_tpu_delegate = None
         try:
-            edge_tpu_delegate = load_delegate('libedgetpu.so.1.0')
+            edge_tpu_delegate = load_delegate('libedgetpu.so.1.0', {"device": "usb"})
+            print("USB TPU found")
         except ValueError:
-            print("No EdgeTPU detected. Falling back to CPU.")
+            try:
+                edge_tpu_delegate = load_delegate('libedgetpu.so.1.0', {"device": "pci:0"})
+                print("PCIe TPU found")
+            except ValueError:
+                print("No EdgeTPU detected. Falling back to CPU.")
         
         if edge_tpu_delegate is None:
             self.interpreter = tflite.Interpreter(
@@ -139,4 +144,4 @@ class RemoteObjectDetector():
             ))
         self.plasma_client.delete([object_id_frame, object_id_detections])
         self.fps.update()
-        return detections
+        return detections