Quellcode durchsuchen

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

Carl Elkins vor 4 Jahren
Ursprung
Commit
a8556a729b
1 geänderte Dateien mit 8 neuen und 3 gelöschten Zeilen
  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