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

Rework face recognition APIs

Browse Source
This commit is contained in:
Nicolas Mowen 2025-02-09 11:06:23 -07:00
parent a42ad7ead9
commit b065d3c645
2 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/api/classification.py
View File

@ -118,6 +118,21 @@ def train_face(request: Request, name: str, body: dict = None):
@router.post("/faces/{name}")
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)), 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(

2
web/src/pages/FaceLibrary.tsx
View File

@ -82,7 +82,7 @@ export default function FaceLibrary() {
const formData = new FormData();
formData.append("file", file);
axios
.post(`faces/${pageToggle}`, formData, {
.post(`faces/${pageToggle}/register`, formData, {
headers: {
"Content-Type": "multipart/form-data",
},