diff --git a/frigate/edgetpu.py b/frigate/edgetpu.py index a5a216deb..dd3592441 100644 --- a/frigate/edgetpu.py +++ b/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 \ No newline at end of file + return detections