github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-07-24 02:40:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1418 from PyWoody/pickle_protocol

Browse Source 
pickle objects are now dumped with pickle.HIGHEST_PROTOCOL
This commit is contained in:
Sefik Ilkin Serengil 2025-01-06 09:29:31 +00:00 committed by GitHub
commit fea8b11cf5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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
if not os.path.exists(datastore_path):
with open(datastore_path, "wb") as f:
pickle.dump([], f)
pickle.dump([], f, pickle.HIGHEST_PROTOCOL)
# Load the representations from the pickle file
with open(datastore_path, "rb") as f:
@ -232,7 +232,7 @@ def find(
if must_save_pickle:
with open(datastore_path, "wb") as f:
pickle.dump(representations, f)
pickle.dump(representations, f, pickle.HIGHEST_PROTOCOL)
if not silent:
logger.info(f"There are now {len(representations)} representations in {file_name}")