mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-04 04:27:42 +03:00
Compare commits
2 Commits
7b511bdebb
...
4f53240822
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4f53240822 | ||
|
|
61282e0754 |
@ -529,7 +529,9 @@ class Embeddings:
|
|||||||
)
|
)
|
||||||
|
|
||||||
# Calculate embedding for new trigger
|
# Calculate embedding for new trigger
|
||||||
embedding = self._calculate_trigger_embedding(trigger, trigger_name, camera.name)
|
embedding = self._calculate_trigger_embedding(
|
||||||
|
trigger, trigger_name, camera.name
|
||||||
|
)
|
||||||
|
|
||||||
Trigger.create(
|
Trigger.create(
|
||||||
camera=camera.name,
|
camera=camera.name,
|
||||||
@ -585,13 +587,13 @@ class Embeddings:
|
|||||||
f"Failed to delete thumbnail for trigger with data {event_id} in {camera}: {e}"
|
f"Failed to delete thumbnail for trigger with data {event_id} in {camera}: {e}"
|
||||||
)
|
)
|
||||||
|
|
||||||
def _calculate_trigger_embedding(self, trigger, trigger_name: str, camera_name: str) -> bytes:
|
def _calculate_trigger_embedding(
|
||||||
|
self, trigger, trigger_name: str, camera_name: str
|
||||||
|
) -> bytes:
|
||||||
"""Calculate embedding for a trigger based on its type and data."""
|
"""Calculate embedding for a trigger based on its type and data."""
|
||||||
if trigger.type == "description":
|
if trigger.type == "description":
|
||||||
logger.debug(f"Generating embedding for trigger description {trigger_name}")
|
logger.debug(f"Generating embedding for trigger description {trigger_name}")
|
||||||
embedding = self.embed_description(
|
embedding = self.embed_description(None, trigger.data, upsert=False)
|
||||||
None, trigger.data, upsert=False
|
|
||||||
)
|
|
||||||
return embedding.astype(np.float32).tobytes()
|
return embedding.astype(np.float32).tobytes()
|
||||||
|
|
||||||
elif trigger.type == "thumbnail":
|
elif trigger.type == "thumbnail":
|
||||||
@ -611,9 +613,7 @@ class Embeddings:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
with open(
|
with open(
|
||||||
os.path.join(
|
os.path.join(TRIGGER_DIR, camera_name, f"{trigger.data}.webp"),
|
||||||
TRIGGER_DIR, camera_name, f"{trigger.data}.webp"
|
|
||||||
),
|
|
||||||
"rb",
|
"rb",
|
||||||
) as f:
|
) as f:
|
||||||
thumbnail = f.read()
|
thumbnail = f.read()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user