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

Use datetime as ID

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-08 08:15:32 -06:00
parent 77aef4c192
commit 325a150cc4
2 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/api/classification.py
View File

@ -1,10 +1,9 @@
"""Object classification APIs.""" """Object classification APIs."""
import datetime
import logging import logging
import os import os
import random
import shutil import shutil
import string
import cv2 import cv2
from fastapi import APIRouter, Depends, Request, UploadFile from fastapi import APIRouter, Depends, Request, UploadFile
@ -120,8 +119,7 @@ def train_face(request: Request, name: str, body: dict = None):
) )
sanitized_name = sanitize_filename(name) sanitized_name = sanitize_filename(name)
rand_id = "".join(random.choices(string.ascii_lowercase + string.digits, k=6)) new_name = f"{sanitized_name}-{datetime.datetime.now().timestamp()}.webp"
new_name = f"{sanitized_name}-{rand_id}.webp"
new_file_folder = os.path.join(FACE_DIR, f"{sanitized_name}") new_file_folder = os.path.join(FACE_DIR, f"{sanitized_name}")
if not os.path.exists(new_file_folder): if not os.path.exists(new_file_folder):

10
frigate/data_processing/real_time/face.py
View File

@ -5,9 +5,7 @@ import datetime
import json import json
import logging import logging
import os import os
import random
import shutil import shutil
import string
from typing import Optional from typing import Optional
import cv2 import cv2
@ -343,11 +341,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
return {"success": True, "score": score, "face_name": sub_label} return {"success": True, "score": score, "face_name": sub_label}
elif topic == EmbeddingsRequestEnum.register_face.value: elif topic == EmbeddingsRequestEnum.register_face.value:
rand_id = "".join(
random.choices(string.ascii_lowercase + string.digits, k=6)
)
label = request_data["face_name"] label = request_data["face_name"]
id = f"{label}-{rand_id}"
if request_data.get("cropped"): if request_data.get("cropped"):
thumbnail = request_data["image"] thumbnail = request_data["image"]
@ -376,7 +370,9 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
# write face to library # write face to library
folder = os.path.join(FACE_DIR, label) folder = os.path.join(FACE_DIR, label)
file = os.path.join(folder, f"{id}.webp") file = os.path.join(
folder, f"{label}_{datetime.datetime.now().timestamp()}.webp"
)
os.makedirs(folder, exist_ok=True) os.makedirs(folder, exist_ok=True)
# save face image # save face image