jeevana commited on
Commit
a29cb70
·
1 Parent(s): 67f4aad

similarity code is updated07

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -91,8 +91,9 @@ def get_similarity(img1, img2):
91
 
92
  # YOUR CODE HERE, return similarity measure using your model
93
  feature_net.eval()
94
-
95
- output1,output2 = feature_net(face1,face2)
 
96
  euclidean_distance = F.pairwise_distance(output1, output2)
97
  #pairwise - more distance means less similarity
98
  #cosine similarity - more means more similarity btwn 2 arrays
 
91
 
92
  # YOUR CODE HERE, return similarity measure using your model
93
  feature_net.eval()
94
+ normalized_face1 = F.normalize(face1, p=2, dim=1)
95
+ normalized_face2 = F.normalize(face2, p=2, dim=1)
96
+ output1,output2 = feature_net(normalized_face1,normalized_face2)
97
  euclidean_distance = F.pairwise_distance(output1, output2)
98
  #pairwise - more distance means less similarity
99
  #cosine similarity - more means more similarity btwn 2 arrays