jeevana commited on
Commit
87a4080
·
1 Parent(s): dac0d4d

similarity code is updated10

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