charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix printing directory

Browse Source
This commit is contained in:
Lorenzo 2021-03-22 15:42:25 +01:00
parent ef31ece267
commit 9e0877e150
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monoloco/__init__.py
View File

@ -3,4 +3,4 @@
Open implementation of MonoLoco / MonoLoco++ / MonStereo
"""
__version__ = '0.5.0'
__version__ = '0.5'

4
monoloco/predict.py
View File

@ -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