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

cleanup

Browse Source
This commit is contained in:
Blake Blackshear 2020-09-25 19:00:59 -05:00
parent 574ee2a46f
commit efdcfcef97
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/edgetpu.py
View File

@ -126,8 +126,6 @@ def run_detector(detection_queue, out_events: Dict[str, mp.Event], avg_speed, st
# detect and send the output # detect and send the output
start.value = datetime.datetime.now().timestamp() start.value = datetime.datetime.now().timestamp()
# TODO: what is the overhead for pickling this result vs writing back to shared memory?
# I could try using an Event() and waiting in the other process before looking in memory...
detections = object_detector.detect_raw(input_frame) detections = object_detector.detect_raw(input_frame)
duration = datetime.datetime.now().timestamp()-start.value duration = datetime.datetime.now().timestamp()-start.value
outputs[connection_id]['np'][:] = detections[:] outputs[connection_id]['np'][:] = detections[:]
@ -179,17 +177,11 @@ class RemoteObjectDetector():
detections = [] detections = []
# copy input to shared memory # copy input to shared memory
# TODO: what if I just write it there in the first place?
self.np_shm[:] = tensor_input[:] self.np_shm[:] = tensor_input[:]
self.event.clear() self.event.clear()
self.detection_queue.put(self.name) self.detection_queue.put(self.name)
self.event.wait() self.event.wait()
# if self.result_connection.poll(10):
# raw_detections = self.result_connection.recv()
# else:
# return detections
for d in self.out_np_shm: for d in self.out_np_shm:
if d[1] < threshold: if d[1] < threshold:
break break