From 05013e550cda3d460c7cdd216b27e17971135103 Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Mon, 8 Jan 2024 16:57:41 +0000 Subject: [PATCH] new threshold values for vgg-face tuned on lfw dataset --- deepface/commons/distance.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/deepface/commons/distance.py b/deepface/commons/distance.py index 4ed2247..1744736 100644 --- a/deepface/commons/distance.py +++ b/deepface/commons/distance.py @@ -41,7 +41,12 @@ def findThreshold(model_name: str, distance_metric: str) -> float: base_threshold = {"cosine": 0.40, "euclidean": 0.55, "euclidean_l2": 0.75} thresholds = { - "VGG-Face": {"cosine": 0.40, "euclidean": 0.60, "euclidean_l2": 0.86}, + # "VGG-Face": {"cosine": 0.40, "euclidean": 0.60, "euclidean_l2": 0.86}, # 2622d + "VGG-Face": { + "cosine": 0.68, + "euclidean": 1.17, + "euclidean_l2": 1.17, + }, # 4096d - tuned with LFW "Facenet": {"cosine": 0.40, "euclidean": 10, "euclidean_l2": 0.80}, "Facenet512": {"cosine": 0.30, "euclidean": 23.56, "euclidean_l2": 1.04}, "ArcFace": {"cosine": 0.68, "euclidean": 4.15, "euclidean_l2": 1.13},