diff --git a/web/src/pages/FaceLibrary.tsx b/web/src/pages/FaceLibrary.tsx index 2309fdb5d..153fcfd11 100644 --- a/web/src/pages/FaceLibrary.tsx +++ b/web/src/pages/FaceLibrary.tsx @@ -18,6 +18,8 @@ import { TooltipTrigger, } from "@/components/ui/tooltip"; import useOptimisticState from "@/hooks/use-optimistic-state"; +import { cn } from "@/lib/utils"; +import { FrigateConfig } from "@/types/frigateConfig"; import axios from "axios"; import { useCallback, useEffect, useMemo, useRef, useState } from "react"; import { LuImagePlus, LuTrash2 } from "react-icons/lu"; @@ -189,11 +191,13 @@ export default function FaceLibrary() { } type TrainingGridProps = { + config: FrigateConfig; attemptImages: string[]; faceNames: string[]; onRefresh: () => void; }; function TrainingGrid({ + config, attemptImages, faceNames, onRefresh, @@ -205,6 +209,7 @@ function TrainingGrid({ key={image} image={image} faceNames={faceNames} + threshold={config.face_recognition.threshold} onRefresh={onRefresh} /> ))} @@ -215,9 +220,15 @@ function TrainingGrid({ type FaceAttemptProps = { image: string; faceNames: string[]; + threshold: number; onRefresh: () => void; }; -function FaceAttempt({ image, faceNames, onRefresh }: FaceAttemptProps) { +function FaceAttempt({ + image, + faceNames, + threshold, + onRefresh, +}: FaceAttemptProps) { const data = useMemo(() => { const parts = image.split("-"); @@ -288,7 +299,13 @@ function FaceAttempt({ image, faceNames, onRefresh }: FaceAttemptProps) {