Merge pull request #1334 from kremnik/checkimagefirst

Improvement: check image exists first
serengil · Sep 3, 2024 · 9d15881 · 9d15881
2 parents d3200c2 + b2e3d8d
commit 9d15881
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepface/modules/recognition.py b/deepface/modules/recognition.py
@@ -100,6 +100,10 @@ def find(
     if not os.path.isdir(db_path):
         raise ValueError(f"Passed path {db_path} does not exist!")
 
+    img, _ = image_utils.load_image(img_path)
+    if img is None:
+        raise ValueError(f"Passed image path {img_path} does not exist!")
+
     file_parts = [
         "ds",
         "model",