jeevana commited on
Commit
939cf55
·
1 Parent(s): 1d79fd0

similarity code is updated04

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -99,10 +99,10 @@ def get_similarity(img1, img2):
99
  # Use euclidean similarity to measure the similarity between given two images
100
  euc_similarity = euclidean_distance.item()
101
  print('output1:::',output1)
102
- print('output1:::',output1[0])
103
- print('output1:::',output1[1])
104
- cos_similarity = torch.nn.functional.cosine_similarity(output1[1], output2[1]).item()
105
-
106
  return cos_similarity
107
 
108
 
 
99
  # Use euclidean similarity to measure the similarity between given two images
100
  euc_similarity = euclidean_distance.item()
101
  print('output1:::',output1)
102
+ print('output1[0]:::',output1[0])
103
+ cos_similarity1 = torch.nn.functional.cosine_similarity(output1, output2)
104
+ print('cos_similarity1',cos_similarity1)
105
+ cos_similarity = cos_similarity1.item()
106
  return cos_similarity
107
 
108