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

Rebuild classifier when deleting face image

Browse Source
This commit is contained in:
Weitheng Haw 2025-01-28 16:18:28 +00:00
parent 922cd5b675
commit cb779211f1
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/api/classification.py
View File

@ -134,6 +134,14 @@ def deregister_faces(request: Request, name: str, body: dict = None):
context.delete_face_ids( context.delete_face_ids(
name, map(lambda file: sanitize_filename(file), list_of_ids) name, map(lambda file: sanitize_filename(file), list_of_ids)
) )
context: EmbeddingsContext = request.app.embeddings
context.clear_face_classifier()
return JSONResponse(
content={"success": True, "message": "Successfully deleted faces."},
status_code=200,
)
except Exception as e: except Exception as e:
logger.error(f"Failed to delete face: {str(e)}") logger.error(f"Failed to delete face: {str(e)}")
return JSONResponse( return JSONResponse(
@ -141,11 +149,6 @@ def deregister_faces(request: Request, name: str, body: dict = None):
status_code=500, status_code=500,
) )
return JSONResponse(
content={"success": True, "message": "Successfully deleted faces."},
status_code=200,
)
@router.post("/faces/{name}/create") @router.post("/faces/{name}/create")
def create_face(name: str): def create_face(name: str):
@ -198,11 +201,9 @@ def rename_face(request: Request, name: str, body: dict = None):
) )
try: try:
# Use atomic operation when possible
try: try:
os.rename(old_folder, new_folder) os.rename(old_folder, new_folder)
except OSError: except OSError:
# Fallback to copy+delete if rename fails
shutil.copytree(old_folder, new_folder) shutil.copytree(old_folder, new_folder)
shutil.rmtree(old_folder) shutil.rmtree(old_folder)