diff --git a/deepface/models/demography/Age.py b/deepface/models/demography/Age.py index ca02a92..bbe99f8 100644 --- 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 index dd7dbf9..1b67ba8 100644 --- 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 index 4fd95a3..636a397 100644 --- 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 index bacec14..8e98698 100644 --- 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 index e8c4f8f..03f5e3d 100644 --- a/deepface/models/facial_recognition/Dlib.py +++ b/deepface/models/facial_recognition/Dlib.py @@ -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)