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

pickle objects are now dumped with pickle.HIGHEST_PROTOCOL

Browse Source
This commit is contained in:
Samuel J. Woodward 2025-01-04 12:05:58 -05:00
parent d5ba8fda09
commit 4255b4167f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/modules/recognition.py
View File

@ -149,7 +149,7 @@ def find(
# Ensure the proper pickle file exists # Ensure the proper pickle file exists
if not os.path.exists(datastore_path): if not os.path.exists(datastore_path):
with open(datastore_path, "wb") as f: with open(datastore_path, "wb") as f:
pickle.dump([], f) pickle.dump([], f, pickle.HIGHEST_PROTOCOL)
# Load the representations from the pickle file # Load the representations from the pickle file
with open(datastore_path, "rb") as f: with open(datastore_path, "rb") as f:
@ -232,7 +232,7 @@ def find(
if must_save_pickle: if must_save_pickle:
with open(datastore_path, "wb") as f: with open(datastore_path, "wb") as f:
pickle.dump(representations, f) pickle.dump(representations, f, pickle.HIGHEST_PROTOCOL)
if not silent: if not silent:
logger.info(f"There are now {len(representations)} representations in {file_name}") logger.info(f"There are now {len(representations)} representations in {file_name}")