jeevana commited on
Commit
1d79fd0
·
1 Parent(s): 8ff1942

similarity code is updated03

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -98,7 +98,10 @@ def get_similarity(img1, img2):
98
  #cosine similarity - more means more similarity btwn 2 arrays
99
  # Use euclidean similarity to measure the similarity between given two images
100
  euc_similarity = euclidean_distance.item()
101
- cos_similarity = torch.nn.functional.cosine_similarity(face1, face2).item()
 
 
 
102
 
103
  return cos_similarity
104
 
 
98
  #cosine similarity - more means more similarity btwn 2 arrays
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