forked from Github/frigate
Compare commits
2 Commits
v0.5.0-rc6
...
v0.5.0-rc7
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
0e6eca7cd6 | ||
|
|
91415f7e9d |
@@ -38,9 +38,9 @@ RUN apt -qq update && apt -qq install --no-install-recommends -y \
|
||||
&& apt -qq install --no-install-recommends -y \
|
||||
libedgetpu1-max \
|
||||
## Tensorflow lite (python 3.7 only)
|
||||
&& wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0-cp37-cp37m-linux_x86_64.whl \
|
||||
&& python3.7 -m pip install tflite_runtime-2.1.0-cp37-cp37m-linux_x86_64.whl \
|
||||
&& rm tflite_runtime-2.1.0-cp37-cp37m-linux_x86_64.whl \
|
||||
&& wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
||||
&& python3.7 -m pip install tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
||||
&& rm tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
||||
&& rm -rf /var/lib/apt/lists/* \
|
||||
&& (apt-get autoremove -y; apt-get autoclean -y)
|
||||
|
||||
|
||||
@@ -75,7 +75,6 @@ def run_detector(detection_queue, avg_speed, start):
|
||||
input_frame = plasma_client.get(object_id, timeout_ms=0)
|
||||
|
||||
if input_frame is plasma.ObjectNotAvailable:
|
||||
plasma_client.put(np.zeros((20,6), np.float32), object_id_out)
|
||||
continue
|
||||
|
||||
# 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())
|
||||
self.plasma_client.put(tensor_input, object_id_frame)
|
||||
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:
|
||||
if d[1] < threshold:
|
||||
|
||||
Reference in New Issue
Block a user