change model recognition

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 faceViT4.pth filter=lfs diff=lfs merge=lfs -text
 faceNet6.pth filter=lfs diff=lfs merge=lfs -text
+faceNet_update_transformation.pth filter=lfs diff=lfs merge=lfs -text

app.py

@@ -11,7 +11,7 @@ mtcnn = MTCNN(keep_all=True, device='cuda' if torch.cuda.is_available() else 'cp
 
 # Load the pre-trained FaceNet model
 facenet = InceptionResnetV1(pretrained='vggface2').eval().to('cuda' if torch.cuda.is_available() else 'cpu')
-model_path = r'faceNet6.pth'
+model_path = r'faceNet_update_transformation.pth'
 model_state_dict = torch.load(model_path)
 facenet.load_state_dict(model_state_dict)
 facenet.eval()  # Set the model to evaluation mode
@@ -72,10 +72,10 @@ def process_images(image1, image2):
     embedding1 = l2_normalize(embedding1)
     embedding2 = l2_normalize(embedding2)
     
-    distance, is_match = compare_faces(embedding1, embedding2, threshold=0.2)
+    distance, is_match = compare_faces(embedding1, embedding2, threshold=0.25)
     
-    # Calculate confidence
-    confidence = max(0.0, 1.0 - distance / 1.0)  # Ensure confidence is between 0 and 1
+    # # Calculate confidence
+    # confidence = max(0.0, 1.0 - distance / 1.0)  # Ensure confidence is between 0 and 1
     
     end_time = time.time()
     inference_time = end_time - start_time
@@ -84,7 +84,7 @@ def process_images(image1, image2):
     image1_with_box = draw_bounding_box(image1, box1)
     image2_with_box = draw_bounding_box(image2, box2)
     
-    result = f"Distance: {distance:.2f}\nMatch: {is_match}\nConfidence: {confidence:.2f}\nInference time: {inference_time:.2f} seconds"
+    result = f"Distance: {distance:.2f}\nMatch: {is_match}\nInference time: {inference_time:.2f} seconds"
     
     return [image1_with_box, image2_with_box], result
 

faceNet_update_transformation.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86b4b567798373e423655892a9a377038d2cfae87bbb073d3d9ae83b93a94081
+size 112028666