diff --git a/monoloco/__init__.py b/monoloco/__init__.py index fdbea92..5d6032a 100644 --- a/monoloco/__init__.py +++ b/monoloco/__init__.py @@ -3,4 +3,4 @@ Open implementation of MonoLoco / MonoLoco++ / MonStereo """ -__version__ = '0.5.0' +__version__ = '0.5' diff --git a/monoloco/predict.py b/monoloco/predict.py index 37a09c7..c0a8fef 100644 --- a/monoloco/predict.py +++ b/monoloco/predict.py @@ -49,7 +49,7 @@ def download_checkpoints(args): dic_models = {'keypoints': pifpaf_model} if not os.path.exists(pifpaf_model): import gdown - LOG.info("Downloading OpenPifPaf model in %s".format(torch_dir)) + LOG.info(f'Downloading OpenPifPaf model in {torch_dir}') gdown.download(OPENPIFPAF_MODEL, pifpaf_model, quiet=False) if args.mode == 'keypoints': @@ -68,7 +68,7 @@ def download_checkpoints(args): if not os.path.exists(model): import gdown - LOG.info("Downloading model in %s".format(torch_dir)) + LOG.info(f'Downloading model (modality: {args.mode}) in {torch_dir}') gdown.download(path, model, quiet=False) return dic_models