From afb36a5ce969b58c7fbbe595358aef08fefea91d Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Fri, 4 Oct 2024 15:59:00 -0500 Subject: [PATCH] fix similarity search --- frigate/embeddings/embeddings.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index a1f0b9686..90e97efa7 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -132,7 +132,7 @@ class Embeddings: def search_thumbnail( self, query: Union[Event, str], limit=10 ) -> List[Tuple[str, float]]: - if isinstance(query, Event): + if query.__class__ == Event: cursor = self.db.execute_sql( """ SELECT thumbnail_embedding FROM vec_thumbnails WHERE id = ? @@ -151,9 +151,9 @@ class Embeddings: thumbnail = base64.b64decode(query.thumbnail) self.upsert_thumbnail(query.id, thumbnail) image = Image.open(io.BytesIO(thumbnail)).convert("RGB") - query = self.clip_embedding([image])[0] - - query_embedding = self.clip_embedding([query])[0] + query_embedding = self.clip_embedding([image])[0] + else: + query_embedding = self.clip_embedding([query])[0] results = self.db.execute_sql( """