|
@@ -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
|