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

if the detections dont come back in 10s, give up

Browse Source
This commit is contained in:
Blake Blackshear 2020-03-01 20:32:32 -06:00
parent c80137e059
commit 8e9110f42e
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/edgetpu.py
View File

@ -75,7 +75,6 @@ def run_detector(detection_queue, avg_speed, start):
input_frame = plasma_client.get(object_id, timeout_ms=0) input_frame = plasma_client.get(object_id, timeout_ms=0)
if input_frame is plasma.ObjectNotAvailable: if input_frame is plasma.ObjectNotAvailable:
plasma_client.put(np.zeros((20,6), np.float32), object_id_out)
continue continue
# detect and put the output in the plasma store # detect and put the output in the plasma store
@ -124,7 +123,11 @@ class RemoteObjectDetector():
object_id_detections = plasma.ObjectID(hashlib.sha1(str.encode(f"out-{now}")).digest()) object_id_detections = plasma.ObjectID(hashlib.sha1(str.encode(f"out-{now}")).digest())
self.plasma_client.put(tensor_input, object_id_frame) self.plasma_client.put(tensor_input, object_id_frame)
self.detection_queue.put(now) self.detection_queue.put(now)
raw_detections = self.plasma_client.get(object_id_detections) raw_detections = self.plasma_client.get(object_id_detections, timeout_ms=10000)
if raw_detections is plasma.ObjectNotAvailable:
self.plasma_client.delete([object_id_frame])
return detections
for d in raw_detections: for d in raw_detections:
if d[1] < threshold: if d[1] < threshold: