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

Update path handling: fixes

Browse Source
This commit is contained in:
kremnik 2024-08-12 03:18:40 +03:00
parent 0bb94a1a11
commit d4a544f012
5 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/models/demography/Age.py
View File

@ -70,7 +70,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/age_model_weights.h5") output = os.path.join(home, ".deepface/weights/age_model_weights.h5")
if not os.path.isfile(output): if not os.path.isfile(output):
logger.info("age_model_weights.h5 will be downloaded...") logger.info(f"{os.path.basename(output)} will be downloaded...")
gdown.download(url, output, quiet=False) gdown.download(url, output, quiet=False)
age_model.load_weights(output) age_model.load_weights(output)

2
deepface/models/demography/Emotion.py
View File

@ -100,7 +100,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/facial_expression_model_weights.h5") output = os.path.join(home, ".deepface/weights/facial_expression_model_weights.h5")
if not os.path.isfile(output): if not os.path.isfile(output):
logger.info("Facial_expression_model_weights.h5 will be downloaded...") logger.info(f"{os.path.basename(output)} will be downloaded...")
gdown.download(url, output, quiet=False) gdown.download(url, output, quiet=False)
model.load_weights(output) model.load_weights(output)

2
deepface/models/demography/Gender.py
View File

@ -77,7 +77,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/gender_model_weights.h5") output = os.path.join(home, ".deepface/weights/gender_model_weights.h5")
if not os.path.isfile(output): if not os.path.isfile(output):
logger.info("gender_model_weights.h5 will be downloaded...") logger.info(f"{os.path.basename(output)} will be downloaded...")
gdown.download(url, output, quiet=False) gdown.download(url, output, quiet=False)
gender_model.load_weights(output) gender_model.load_weights(output)

2
deepface/models/demography/Race.py
View File

@ -74,7 +74,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/race_model_single_batch.h5") output = os.path.join(home, ".deepface/weights/race_model_single_batch.h5")
if not os.path.isfile(output): if not os.path.isfile(output):
logger.info("race_model_single_batch.h5 will be downloaded...") logger.info(f"{os.path.basename(output)} will be downloaded...")
gdown.download(url, output, quiet=False) gdown.download(url, output, quiet=False)
race_model.load_weights(output) race_model.load_weights(output)

2
deepface/models/facial_recognition/Dlib.py
View File

@ -73,7 +73,7 @@ class DlibResNet:
# download pre-trained model if it does not exist # download pre-trained model if it does not exist
if not os.path.isfile(weight_file): if not os.path.isfile(weight_file):
logger.info(f"{filename} is going to be downloaded") logger.info(f"{filename} is going to be downloaded")
url = f"http://dlib.net/files/{filename + 'bz2'}" url = f"http://dlib.net/files/{filename + '.bz2'}"
output = weight_file + ".bz2" output = weight_file + ".bz2"
gdown.download(url, output, quiet=False) gdown.download(url, output, quiet=False)