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

Face recognition UI improvements (#16422)

Browse Source 
* Rework face recognition APIs

* Fix error message on cancel

* Add ability to create new face library
This commit is contained in:
Nicolas Mowen
2025-02-09 13:22:25 -06:00
committed by GitHub
parent c58d2add37
commit 81a56549da
4 changed files with 163 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/api/classification.py
+18 -1
View File
@@ -117,7 +117,24 @@ def train_face(request: Request, name: str, body: dict = None):
)
@router.post("/faces/{name}")
@router.post("/faces/{name}/create")
async def create_face(request: Request, name: str, file: UploadFile):
if not request.app.frigate_config.face_recognition.enabled:
return JSONResponse(
status_code=400,
content={"message": "Face recognition is not enabled.", "success": False},
)
os.makedirs(
sanitize_filename(os.path.join(FACE_DIR, name.replace(" ", "_"))), exist_ok=True
)
return JSONResponse(
status_code=200,
content={"success": False, "message": "Successfully created face folder."},
)
@router.post("/faces/{name}/register")
async def register_face(request: Request, name: str, file: UploadFile):
if not request.app.frigate_config.face_recognition.enabled:
return JSONResponse(