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

similarity code is updated09

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -92,8 +92,8 @@ 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, p=2, dim=0)
96
- normalized_face2 = F.normalize(face2, p=2, 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
 
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