Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-07 02:18:07 +01:00
Code Issues Projects Releases Wiki Activity

Face recognition bug fixes (#17401)

Browse Source 
* Simplify normalization

* Fix confidence check
This commit is contained in:
Nicolas Mowen
2025-03-26 18:43:10 -06:00
committed by GitHub
parent a37f804469
commit ff34739f40
2 changed files with 3 additions and 4 deletions
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.const import MODEL_CACHE_DIR
from frigate.embeddings.onnx.facenet import ArcfaceEmbedding
from frigate.embeddings.onnx.face_embedding import ArcfaceEmbedding
logger = logging.getLogger(__name__)
@@ -329,7 +329,7 @@ class ArcFaceRecognizer(FaceRecognizer):
cosine_similarity = dot_product / (magnitude_A * magnitude_B)
confidence = self.similarity_to_confidence(cosine_similarity)
if cosine_similarity > score:
if confidence > score:
score = confidence
label = name