From afc668fe0a1b035bb76cba526d73bd4e8421aaa3 Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Sun, 13 Oct 2024 16:14:24 -0500 Subject: [PATCH] more robust key check and warning message --- frigate/embeddings/functions/onnx.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index 6744fae5c..574822d59 100644 --- a/frigate/embeddings/functions/onnx.py +++ b/frigate/embeddings/functions/onnx.py @@ -190,9 +190,11 @@ class GenericONNXEmbedding: if key in input_names: onnx_inputs[key].append(value[0]) - for key in onnx_inputs.keys(): - if onnx_inputs[key]: + for key in input_names: + if onnx_inputs.get(key): onnx_inputs[key] = np.stack(onnx_inputs[key]) + else: + logger.warning(f"Expected input '{key}' not found in onnx_inputs") embeddings = self.runner.run(onnx_inputs)[0] return [embedding for embedding in embeddings]