administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-05 13:07:44 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix confidence check

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-26 17:35:57 -06:00
parent 3a80490c2e
commit a2ff23ee73
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/data_processing/common/face/model.py
View File

@ -10,7 +10,7 @@ from scipy import stats
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.const import MODEL_CACHE_DIR from frigate.const import MODEL_CACHE_DIR
from frigate.embeddings.onnx.facenet import ArcfaceEmbedding from frigate.embeddings.onnx.face_embedding import ArcfaceEmbedding
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -329,7 +329,7 @@ class ArcFaceRecognizer(FaceRecognizer):
cosine_similarity = dot_product / (magnitude_A * magnitude_B) cosine_similarity = dot_product / (magnitude_A * magnitude_B)
confidence = self.similarity_to_confidence(cosine_similarity) confidence = self.similarity_to_confidence(cosine_similarity)
if cosine_similarity > score: if confidence > score:
score = confidence score = confidence
label = name label = name