Spaces:
Runtime error
Runtime error
Commit ·
5f4efdb
1
Parent(s): 074f9ea
flatten input to cosine distance
Browse files- faceaiapi.py +1 -1
faceaiapi.py
CHANGED
|
@@ -80,7 +80,7 @@ def generate_embeddings(fpath='./encode/'):
|
|
| 80 |
|
| 81 |
|
| 82 |
def compare(embedding_known,embedding_unknown,limit=0.45):
|
| 83 |
-
dist=spatial.distance.cosine(embedding_known,embedding_unknown)
|
| 84 |
if dist>limit:
|
| 85 |
return 0
|
| 86 |
else:
|
|
|
|
| 80 |
|
| 81 |
|
| 82 |
def compare(embedding_known,embedding_unknown,limit=0.45):
|
| 83 |
+
dist=spatial.distance.cosine(embedding_known.flatten(),embedding_unknown.flatten())
|
| 84 |
if dist>limit:
|
| 85 |
return 0
|
| 86 |
else:
|