similarity code is updated

app/Hackathon_setup/face_recognition.py

@@ -11,6 +11,7 @@ import io
 import os
 import joblib
 import pickle
+import torch.nn.functional as F
 
 # Add more imports if required
 
@@ -89,11 +90,14 @@ def get_similarity(img1, img2):
     ##
 
     # YOUR CODE HERE, return similarity measure using your model
-    features_1 = feature_net(face1)[1]
-    features_2 = feature_net(face2)[1]
-
-            # Use cosine similarity to measure the similarity between given two images
-    dissimilarity = torch.nn.functional.cosine_similarity(features_1, features_2).item()
+    feature_net.eval()
+    output1,output2 = feature_net(face1,face2)
+    euclidean_distance = F.pairwise_distance(output1, output2)
+    #features_1 = feature_net(face1)[1]
+    #features_2 = feature_net(face2)[1]
+
+     # Use euclidean similarity to measure the similarity between given two images
+    dissimilarity = euclidean_distance.item()
 
     return dissimilarity