Spaces:
Sleeping
Sleeping
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()
|