Spaces:
Sleeping
Sleeping
similarity code is updated09
Browse files
app/Hackathon_setup/face_recognition.py
CHANGED
|
@@ -92,8 +92,8 @@ def get_similarity(img1, img2):
|
|
| 92 |
# YOUR CODE HERE, return similarity measure using your model
|
| 93 |
feature_net.eval()
|
| 94 |
print('face1.type', type(face1))
|
| 95 |
-
normalized_face1 = F.normalize(face1,
|
| 96 |
-
normalized_face2 = F.normalize(face2,
|
| 97 |
output1,output2 = feature_net(normalized_face1,normalized_face2)
|
| 98 |
euclidean_distance = F.pairwise_distance(output1, output2)
|
| 99 |
#pairwise - more distance means less similarity
|
|
|
|
| 92 |
# YOUR CODE HERE, return similarity measure using your model
|
| 93 |
feature_net.eval()
|
| 94 |
print('face1.type', type(face1))
|
| 95 |
+
normalized_face1 = F.normalize(face1, dim=0)
|
| 96 |
+
normalized_face2 = F.normalize(face2, dim=0)
|
| 97 |
output1,output2 = feature_net(normalized_face1,normalized_face2)
|
| 98 |
euclidean_distance = F.pairwise_distance(output1, output2)
|
| 99 |
#pairwise - more distance means less similarity
|