Update path handling: fixes

serengil · Aug 12, 2024 · d4a544f · d4a544f
1 parent 0bb94a1
commit d4a544f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/deepface/models/demography/Age.py b/deepface/models/demography/Age.py
@@ -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)

diff --git a/deepface/models/demography/Emotion.py b/deepface/models/demography/Emotion.py
@@ -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)

diff --git a/deepface/models/demography/Gender.py b/deepface/models/demography/Gender.py
@@ -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)

diff --git a/deepface/models/demography/Race.py b/deepface/models/demography/Race.py
@@ -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)

diff --git a/deepface/models/facial_recognition/Dlib.py b/deepface/models/facial_recognition/Dlib.py
@@ -73,7 +73,7 @@ def __init__(self):
         # 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)