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

raise error if refresh db is false and pkl is empty

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-05-18 14:05:46 +01:00
parent 3a1a4a06e3
commit c04e51e472
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/modules/recognition.py
View File

@ -151,6 +151,11 @@ def find(
# Get the list of images on storage # Get the list of images on storage
storage_images = image_utils.list_images(path=db_path) storage_images = image_utils.list_images(path=db_path)
if len(storage_images) == 0 and refresh_database is True:
raise ValueError(f"No item found in {db_path}")
if len(representations) == 0 and refresh_database is False:
raise ValueError(f"Nothing is found in {datastore_path}")
must_save_pickle = False must_save_pickle = False
new_images = [] new_images = []
old_images = [] old_images = []
@ -164,11 +169,7 @@ def find(
# Enforce data consistency amongst on disk images and pickle file # Enforce data consistency amongst on disk images and pickle file
if refresh_database: if refresh_database:
if len(storage_images) == 0:
raise ValueError(f"No item found in {db_path}")
new_images = list(set(storage_images) - set(pickled_images)) # images added to storage new_images = list(set(storage_images) - set(pickled_images)) # images added to storage
old_images = list(set(pickled_images) - set(storage_images)) # images removed from storage old_images = list(set(pickled_images) - set(storage_images)) # images removed from storage
# detect replaced images # detect replaced images