mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-19 01:17:06 +03:00
Rework face recognition APIs
This commit is contained in:
parent
a42ad7ead9
commit
b065d3c645
@ -118,6 +118,21 @@ def train_face(request: Request, name: str, body: dict = None):
|
|||||||
|
|
||||||
|
|
||||||
@router.post("/faces/{name}")
|
@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):
|
async def register_face(request: Request, name: str, file: UploadFile):
|
||||||
if not request.app.frigate_config.face_recognition.enabled:
|
if not request.app.frigate_config.face_recognition.enabled:
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
|
|||||||
@ -82,7 +82,7 @@ export default function FaceLibrary() {
|
|||||||
const formData = new FormData();
|
const formData = new FormData();
|
||||||
formData.append("file", file);
|
formData.append("file", file);
|
||||||
axios
|
axios
|
||||||
.post(`faces/${pageToggle}`, formData, {
|
.post(`faces/${pageToggle}/register`, formData, {
|
||||||
headers: {
|
headers: {
|
||||||
"Content-Type": "multipart/form-data",
|
"Content-Type": "multipart/form-data",
|
||||||
},
|
},
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user