Update path handling: fixes

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

View File

@ -70,7 +70,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/age_model_weights.h5")
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)
age_model.load_weights(output)

View File

@ -100,7 +100,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/facial_expression_model_weights.h5")
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)
model.load_weights(output)

View File

@ -77,7 +77,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/gender_model_weights.h5")
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)
gender_model.load_weights(output)

View File

@ -74,7 +74,7 @@ def load_model(
output = os.path.join(home, ".deepface/weights/race_model_single_batch.h5")
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)
race_model.load_weights(output)

View File

@ -73,7 +73,7 @@ class DlibResNet:
# download pre-trained model if it does not exist
if not os.path.isfile(weight_file):
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"
gdown.download(url, output, quiet=False)