From 181b53a55d5377366ecb99b1d312b286146576b8 Mon Sep 17 00:00:00 2001 From: Sergey Krashevich Date: Mon, 15 May 2023 15:37:34 +0300 Subject: [PATCH] Fix error in parsing DeepStack response JSON and handle cases where predictions field is missing (#6463) --- frigate/detectors/plugins/deepstack.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/frigate/detectors/plugins/deepstack.py b/frigate/detectors/plugins/deepstack.py index 716065903c..9d2c695405 100644 --- a/frigate/detectors/plugins/deepstack.py +++ b/frigate/detectors/plugins/deepstack.py @@ -56,8 +56,11 @@ def detect_raw(self, tensor_input): ) response_json = response.json() detections = np.zeros((20, 6), np.float32) + if response_json.get("predictions") is None: + logger.debug(f"Error in parsing response json: {response_json}") + return detections - for i, detection in enumerate(response_json["predictions"]): + for i, detection in enumerate(response_json.get("predictions")): logger.debug(f"Response: {detection}") if detection["confidence"] < 0.4: logger.debug(f"Break due to confidence < 0.4")