administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-18 17:14:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Change folder structure to clips/genai-requests/{event_id}/{1.jpg}

Browse Source
This commit is contained in:
leccelecce 2024-12-23 13:33:10 +00:00
parent 9e80af344e
commit 0fee2b75a6
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/embeddings/maintainer.py
View File

@ -5,6 +5,7 @@ import logging
import os import os
import threading import threading
from multiprocessing.synchronize import Event as MpEvent from multiprocessing.synchronize import Event as MpEvent
from pathlib import Path
from typing import Optional from typing import Optional
import cv2 import cv2
@ -237,7 +238,11 @@ class EmbeddingMaintainer(threading.Thread):
f"Saving {num_thumbnails} thumbnails for event {event.id}" f"Saving {num_thumbnails} thumbnails for event {event.id}"
) )
for idx, data in enumerate(self.tracked_events[event_id]): Path(
os.path.join(CLIPS_DIR, f"genai-requests/{event.id}")
).mkdir(parents=True, exist_ok=True)
for idx, data in enumerate(self.tracked_events[event_id], 1):
jpg_bytes: bytes = data["thumbnail"] jpg_bytes: bytes = data["thumbnail"]
if jpg_bytes is None: if jpg_bytes is None:
@ -248,7 +253,7 @@ class EmbeddingMaintainer(threading.Thread):
with open( with open(
os.path.join( os.path.join(
CLIPS_DIR, CLIPS_DIR,
f"thumb-{camera}-{event.id}-{idx}.jpg", f"genai-requests/{event.id}/{idx}.jpg",
), ),
"wb", "wb",
) as j: ) as j: