Spaces:
Sleeping
Sleeping
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 |
-
|
| 104 |
-
|
| 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 |
|