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

change text

Browse Source
This commit is contained in:
Lorenzo 2021-05-18 10:45:37 +02:00
parent 0660b97cec
commit 3167612a32
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monoloco/predict.py
View File

@ -61,7 +61,8 @@ def download_checkpoints(args):
pifpaf_model = args.checkpoint pifpaf_model = args.checkpoint
dic_models = {'keypoints': pifpaf_model} dic_models = {'keypoints': pifpaf_model}
if not os.path.exists(pifpaf_model): if not os.path.exists(pifpaf_model):
assert DOWNLOAD is not None, "install gdown to download pifpaf model, or pass it as --checkpoint" assert DOWNLOAD is not None, \
"pip install gdown to download a pifpaf model, or pass the model path as --checkpoint"
LOG.info('Downloading OpenPifPaf model in %s', torch_dir) LOG.info('Downloading OpenPifPaf model in %s', torch_dir)
DOWNLOAD(OPENPIFPAF_MODEL, pifpaf_model, quiet=False) DOWNLOAD(OPENPIFPAF_MODEL, pifpaf_model, quiet=False)
@ -86,7 +87,8 @@ def download_checkpoints(args):
dic_models[args.mode] = model dic_models[args.mode] = model
if not os.path.exists(model): if not os.path.exists(model):
os.makedirs(torch_dir, exist_ok=True) os.makedirs(torch_dir, exist_ok=True)
assert DOWNLOAD is not None, "pip install gdown to download monoloco model, or pass it as --model" assert DOWNLOAD is not None, \
"pip install gdown to download a monoloco model, or pass the model path as --model"
LOG.info('Downloading model in %s', torch_dir) LOG.info('Downloading model in %s', torch_dir)
DOWNLOAD(path, model, quiet=False) DOWNLOAD(path, model, quiet=False)
return dic_models return dic_models