administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-27 09:07:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Cleanup

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-13 11:47:47 -06:00
parent c688c89386
commit af7d09300e
1 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
frigate/detectors/plugins/zmq_ipc.py
View File

@ -73,8 +73,9 @@ class ZmqIpcDetector(DetectionApi):
# Apply timeouts and linger so calls don't block indefinitely # Apply timeouts and linger so calls don't block indefinitely
self._socket.setsockopt(zmq.RCVTIMEO, self._request_timeout_ms) self._socket.setsockopt(zmq.RCVTIMEO, self._request_timeout_ms)
self._socket.setsockopt(zmq.SNDTIMEO, self._request_timeout_ms) self._socket.setsockopt(zmq.SNDTIMEO, self._request_timeout_ms)
#self._socket.setsockopt(zmq.LINGER, self._linger_ms) self._socket.setsockopt(zmq.LINGER, self._linger_ms)
logger.info(f"ZMQ detector connecting to {self._endpoint}")
logger.debug(f"ZMQ detector connecting to {self._endpoint}")
self._socket.connect(self._endpoint) self._socket.connect(self._endpoint)
def _build_header(self, tensor_input: np.ndarray) -> bytes: def _build_header(self, tensor_input: np.ndarray) -> bytes:
@ -114,25 +115,15 @@ class ZmqIpcDetector(DetectionApi):
payload_bytes = memoryview(tensor_input.tobytes(order="C")) payload_bytes = memoryview(tensor_input.tobytes(order="C"))
# Send request # Send request
print("sending multipart")
self._socket.send_multipart([header_bytes, payload_bytes]) self._socket.send_multipart([header_bytes, payload_bytes])
print("sent multipart")
# Receive reply # Receive reply
print("looking for response")
reply_frames = self._socket.recv_multipart() reply_frames = self._socket.recv_multipart()
detections = self._decode_response(reply_frames) detections = self._decode_response(reply_frames)
print(f"got response of first item {detections[0]}")
# Ensure output shape and dtype are exactly as expected # Ensure output shape and dtype are exactly as expected
if detections.shape != (20, 6) or detections.dtype != np.float32:
try:
detections = np.asarray(detections, dtype=np.float32)
detections = detections.reshape((20, 6))
except Exception: # noqa: BLE001
logger.warning(
f"ZMQ detector coerced unexpected response shape {detections.shape}"
)
detections = self._zero_result
return detections return detections
except zmq.Again: except zmq.Again: