From a97b708ed12becc5029c2c3c3b4007f527130389 Mon Sep 17 00:00:00 2001
From: jiang <jiang@bonus>
Date: Wed, 18 Sep 2024 15:42:50 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BB=A3=E7=A0=81=E4=B8=8E?=
 =?UTF-8?q?=E6=96=87=E6=A1=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 FaceRegWeb5.2/FaceRegWeb/process.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/FaceRegWeb5.2/FaceRegWeb/process.py b/FaceRegWeb5.2/FaceRegWeb/process.py
index f254d1e..81fe824 100644
--- a/FaceRegWeb5.2/FaceRegWeb/process.py
+++ b/FaceRegWeb5.2/FaceRegWeb/process.py
@@ -20,9 +20,10 @@ from models import FaceRecoger, FaceBoxesV2, Landmark5er, FaceAlign, QualityOfCl
 so = onnxruntime.SessionOptions()
 so.log_severity_level = 3  # 0=VERBOSE, 1=INFO, 2=WARNING, 3=ERROR, 4=FATAL
 
-# # 获取workers
-# if "NUM_WORKERS" not in os.environ:
-#     raise RuntimeError("Environment variable NUM_WORKERS is required but not set.")
+
+# 获取workers
+if "NUM_WORKERS" not in os.environ:
+    raise RuntimeError("Environment variable NUM_WORKERS is required but not set.")
 NUM_WORKERS = int(os.getenv("NUM_WORKERS", 10))
 
 # max readers 
@@ -679,7 +680,7 @@ class FaceHelper:
             # Calculate the Euclidean distance between features
             distance = calculate_euclidean_distance(features_first, features_second)
             # Compare distance with threshold
-            return distance >= 0.75
+            return distance >= self.sim_threshold
 
         except Exception as e:
             logger.error(f"Error occurred: {e}")