Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Added support for PCIe TPU, as well as USB

Browse Source 
Also added message showing which found
This commit is contained in:
Carl Elkins 2020-08-29 22:42:41 +00:00 committed by Blake Blackshear
parent 068df3ef2d
commit a8556a729b
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
frigate/edgetpu.py
View File

@ -31,9 +31,14 @@ class ObjectDetector():
def __init__(self): def __init__(self):
edge_tpu_delegate = None edge_tpu_delegate = None
try: 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: 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: if edge_tpu_delegate is None:
self.interpreter = tflite.Interpreter( self.interpreter = tflite.Interpreter(
@ -139,4 +144,4 @@ class RemoteObjectDetector():
)) ))
self.plasma_client.delete([object_id_frame, object_id_detections]) self.plasma_client.delete([object_id_frame, object_id_detections])
self.fps.update() self.fps.update()
return detections return detections