github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-08 12:35:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

distance in find function implemented

Browse Source
This commit is contained in:
Şefik Serangil 2020-05-25 17:23:15 +03:00
parent c67ae7fe6a
commit 4440d82d44
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
deepface/DeepFace.py
View File

@ -435,7 +435,16 @@ def find(img_path, db_path
distances = []
for index, instance in df.iterrows():
source_representation = instance["representation"]
distance = dst.findCosineDistance(source_representation, target_representation)
if distance_metric == 'cosine':
distance = dst.findCosineDistance(source_representation, target_representation)
elif distance_metric == 'euclidean':
distance = dst.findEuclideanDistance(source_representation, target_representation)
elif distance_metric == 'euclidean_l2':
distance = dst.findEuclideanDistance(dst.l2_normalize(source_representation), dst.l2_normalize(target_representation))
else:
raise ValueError("Invalid distance_metric passed - ", distance_metric)
distances.append(distance)
df["distance"] = distances