From e3cc943b0d57f52d8cc96fa1609e997f493248f9 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 22 Oct 2024 12:41:04 -0600 Subject: [PATCH] Small tweaks --- frigate/config/semantic_search.py | 2 +- frigate/embeddings/functions/onnx.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/frigate/config/semantic_search.py b/frigate/config/semantic_search.py index 154423bae..8ed291b1e 100644 --- a/frigate/config/semantic_search.py +++ b/frigate/config/semantic_search.py @@ -13,7 +13,7 @@ class FaceRecognitionConfig(FrigateBaseModel): default=0.8, title="Face similarity score required to be considered a match." ) min_area: int = Field( - default=1000, title="Min area of face box to consider running face recognition." + default=500, title="Min area of face box to consider running face recognition." ) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index bbffbfb0e..274614a1b 100644 --- a/frigate/embeddings/functions/onnx.py +++ b/frigate/embeddings/functions/onnx.py @@ -196,7 +196,7 @@ class GenericONNXEmbedding: # handle images larger than input size width, height = pil.size - if width > 112 or height > 112: + if width != 112 or height != 112: if width > height: new_height = int(((height / width) * 112) // 4 * 4) pil = pil.resize((112, new_height))