Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-04 13:47:37 +02:00
Code Issues Projects Releases Wiki Activity

Cleanup

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-19 08:39:30 -06:00
parent 98a326f025
commit ea78cced54
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/api/classification.py
View File

@ -207,7 +207,7 @@ async def recognize_face(request: Request, file: UploadFile):
) )
context: EmbeddingsContext = request.app.embeddings context: EmbeddingsContext = request.app.embeddings
result = context.recognize_face(name, await file.read()) result = context.recognize_face(await file.read())
return JSONResponse( return JSONResponse(
status_code=200 if result.get("success", True) else 400, status_code=200 if result.get("success", True) else 400,
content=result, content=result,

3
frigate/embeddings/__init__.py
View File

@ -197,11 +197,10 @@ class EmbeddingsContext:
}, },
) )
def recognize_face(self, face_name: str, image_data: bytes) -> dict[str, any]: def recognize_face(self, image_data: bytes) -> dict[str, any]:
return self.requestor.send_data( return self.requestor.send_data(
EmbeddingsRequestEnum.recognize_face.value, EmbeddingsRequestEnum.recognize_face.value,
{ {
"face_name": face_name,
"image": base64.b64encode(image_data).decode("ASCII"), "image": base64.b64encode(image_data).decode("ASCII"),
}, },
) )