From 0ed16ed293bad504feb6c2a9630b9d3caca827ea Mon Sep 17 00:00:00 2001 From: Chirayu Rai Date: Tue, 15 Jul 2025 17:06:17 -0700 Subject: [PATCH] Added guard clause for empty inference reponse --- frigate/detectors/plugins/degirum.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/detectors/plugins/degirum.py b/frigate/detectors/plugins/degirum.py index 7179e1554..07f81e74d 100644 --- a/frigate/detectors/plugins/degirum.py +++ b/frigate/detectors/plugins/degirum.py @@ -88,9 +88,9 @@ class DGDetector(DetectionApi): logger.debug("Prediction generator was called") with self.dg_model as model: while 1: - logger.debug(f"q size before calling get: {self._queue.qsize()}") + logger.info(f"q size before calling get: {self._queue.qsize()}") data = self._queue.get(block=True) - logger.debug(f"q size after calling get: {self._queue.qsize()}") + logger.info(f"q size after calling get: {self._queue.qsize()}") logger.debug( f"Data we're passing into model predict: {data}, shape of data: {data.shape}" ) @@ -113,7 +113,7 @@ class DGDetector(DetectionApi): res = next(self.prediction) # If we have an empty prediction, return immediately - if len(res.results[0]) == 0: + if len(res.results) == 0 or len(res.results[0]) == 0: return detections i = 0