diff --git a/ppgan/apps/first_order_predictor.py b/ppgan/apps/first_order_predictor.py index c0f6dded049076c4946496259d1bafa4028e2bdb..7fe981a4718f356d19b714e808aef0989ccfa857 100644 --- a/ppgan/apps/first_order_predictor.py +++ b/ppgan/apps/first_order_predictor.py @@ -81,6 +81,8 @@ class FirstOrderPredictor(BasePredictor): weight_path = get_path_from_url(vox_cpk_weight_url) self.weight_path = weight_path + if not os.path.exists(output): + os.makedirs(output) self.output = output self.relative = relative self.adapt_scale = adapt_scale diff --git a/ppgan/utils/download.py b/ppgan/utils/download.py index 49e1d061a4c76b409bc2afe4e11e9c0d5da2db10..e9411a29db9bef05f7d487699e27731e86e40a44 100644 --- a/ppgan/utils/download.py +++ b/ppgan/utils/download.py @@ -111,7 +111,7 @@ def _download(url, path, md5sum=None): raise RuntimeError("Download from {} failed. " "Retry limit reached".format(url)) - logger.info("Downloading {} from {}".format(fname, url)) + logger.info("Downloading {} from {} to {}".format(fname, url, fullname)) req = requests.get(url, stream=True) if req.status_code != 200: