administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 15:45:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix similarity search

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-04 15:59:00 -05:00
parent fc242e2001
commit afb36a5ce9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/embeddings/embeddings.py
View File

@ -132,7 +132,7 @@ class Embeddings:
def search_thumbnail( def search_thumbnail(
self, query: Union[Event, str], limit=10 self, query: Union[Event, str], limit=10
) -> List[Tuple[str, float]]: ) -> List[Tuple[str, float]]:
if isinstance(query, Event): if query.__class__ == Event:
cursor = self.db.execute_sql( cursor = self.db.execute_sql(
""" """
SELECT thumbnail_embedding FROM vec_thumbnails WHERE id = ? SELECT thumbnail_embedding FROM vec_thumbnails WHERE id = ?
@ -151,9 +151,9 @@ class Embeddings:
thumbnail = base64.b64decode(query.thumbnail) thumbnail = base64.b64decode(query.thumbnail)
self.upsert_thumbnail(query.id, thumbnail) self.upsert_thumbnail(query.id, thumbnail)
image = Image.open(io.BytesIO(thumbnail)).convert("RGB") image = Image.open(io.BytesIO(thumbnail)).convert("RGB")
query = self.clip_embedding([image])[0] query_embedding = self.clip_embedding([image])[0]
else:
query_embedding = self.clip_embedding([query])[0] query_embedding = self.clip_embedding([query])[0]
results = self.db.execute_sql( results = self.db.execute_sql(
""" """