From 7f8186278d5a5b945b5dc497847f162f43cbf63e Mon Sep 17 00:00:00 2001 From: serengil Date: Fri, 4 Dec 2020 14:50:40 +0300 Subject: [PATCH] .jpg .JPG problem --- deepface/DeepFace.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index f38c7a5..6c7e3c4 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -408,12 +408,12 @@ def find(img_path, db_path, model_name ='VGG-Face', distance_metric = 'cosine', for r, d, f in os.walk(db_path): # r=root, d=directories, f = files for file in f: - if ('.jpg' in file): + if ('.jpg' in file.lower()) or ('.png' in file.lower()): exact_path = r + "/" + file employees.append(exact_path) if len(employees) == 0: - raise ValueError("There is no image in ", db_path," folder!") + raise ValueError("There is no image in ", db_path," folder! Validate .jpg or .png files exist in this path.") #------------------------ #find representations for db images