jeevana commited on
Commit
b3705c6
·
1 Parent(s): 2c6e2c1

face similarity

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -94,15 +94,14 @@ def get_similarity(img1, img2):
94
  print('face1.type', type(face1))
95
 
96
  output1,output2 = feature_net(face1,face2)
97
- normalized_face1 = F.normalize(output1, dim=1)
98
- normalized_face2 = F.normalize(output2, dim=1)
99
- euclidean_distance = F.pairwise_distance(normalized_face1, normalized_face2)
 
100
  #pairwise - more distance means less similarity
101
  #cosine similarity - more means more similarity btwn 2 arrays
102
  # Use euclidean similarity to measure the similarity between given two images
103
  euc_similarity = euclidean_distance.item()
104
- # print('output1:::',output1)
105
- # print('output1[0]:::',output1[0])
106
  # cos_similarity1 = torch.nn.functional.cosine_similarity(output1, output2)
107
  # print('cos_similarity1',cos_similarity1)
108
  # cos_similarity = cos_similarity1.item()
 
94
  print('face1.type', type(face1))
95
 
96
  output1,output2 = feature_net(face1,face2)
97
+ #normalized_face1 = F.normalize(output1, dim=1)
98
+ #normalized_face2 = F.normalize(output2, dim=1)
99
+ #euclidean_distance = F.pairwise_distance(normalized_face1, normalized_face2)
100
+ euclidean_distance = F.pairwise_distance(face1, face2)
101
  #pairwise - more distance means less similarity
102
  #cosine similarity - more means more similarity btwn 2 arrays
103
  # Use euclidean similarity to measure the similarity between given two images
104
  euc_similarity = euclidean_distance.item()
 
 
105
  # cos_similarity1 = torch.nn.functional.cosine_similarity(output1, output2)
106
  # print('cos_similarity1',cos_similarity1)
107
  # cos_similarity = cos_similarity1.item()