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

Use new endpoint to create face directory

Browse Source
This commit is contained in:
Weitheng Haw 2025-01-28 12:14:50 +00:00
parent 400ee3401f
commit 2998771110
2 changed files with 20 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
frigate/api/classification.py
View File

@ -125,3 +125,20 @@ def deregister_faces(request: Request, name: str, body: dict = None):
content=({"success": True, "message": "Successfully deleted faces."}),
status_code=200,
)
@router.post("/faces/{name}/create")
def create_face(name: str):
"""Create a new face directory without requiring an image."""
folder = os.path.join(FACE_DIR, name)
if os.path.exists(folder):
return JSONResponse(
status_code=400,
content={"message": f"Face '{name}' already exists", "success": False},
)
os.makedirs(folder, exist_ok=True)
return JSONResponse(
status_code=200,
content={"message": "Successfully created face", "success": True},
)

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

@ -133,25 +133,8 @@ export default function FaceLibrary() {
setIsCreatingFace(true);
try {
const canvas = document.createElement('canvas');
canvas.width = 1;
canvas.height = 1;
const ctx = canvas.getContext('2d');
if (ctx) {
ctx.fillStyle = 'white';
ctx.fillRect(0, 0, 1, 1);
}
const base64Image = canvas.toDataURL('image/webp').split(',')[1];
const formData = new FormData();
formData.append('cropped', 'true');
formData.append('image', base64Image);
const resp = await axios.post(`/faces/${newFaceName}`, {
cropped: true,
image: base64Image
});
// Create a directory by making a POST request without a file
const resp = await axios.post(`/faces/${newFaceName}/create`);
if (resp.status === 200) {
setNewFaceDialog(false);
@ -182,26 +165,8 @@ export default function FaceLibrary() {
setIsRenaming(true);
try {
// Create a 1x1 transparent WebP image
const canvas = document.createElement('canvas');
canvas.width = 1;
canvas.height = 1;
const ctx = canvas.getContext('2d');
if (ctx) {
ctx.fillStyle = 'white';
ctx.fillRect(0, 0, 1, 1);
}
// Convert to base64
const base64Image = canvas.toDataURL('image/webp').split(',')[1];
await axios.post(`/faces/${renameData.newName}/create`);
// Create new face directory
await axios.post(`/faces/${renameData.newName}`, {
cropped: true,
image: base64Image
});
// Copy existing images
const oldFaceImages = faceData[renameData.oldName] || [];
for (const image of oldFaceImages) {
const response = await fetch(`${baseUrl}clips/faces/${renameData.oldName}/${image}`);