From 5ca3f995d58f13da8589669ac3ca4f01a107c29c Mon Sep 17 00:00:00 2001 From: Jia Guo Date: Sat, 19 Jun 2021 22:37:11 +0800 Subject: [PATCH] create dir after model downloaded --- python-package/insightface/utils/storage.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/python-package/insightface/utils/storage.py b/python-package/insightface/utils/storage.py index 578d311..88d6cf7 100644 --- a/python-package/insightface/utils/storage.py +++ b/python-package/insightface/utils/storage.py @@ -11,23 +11,19 @@ def download(sub_dir, name, force=False, root='~/.insightface'): dir_path = os.path.join(_root, sub_dir, name) if osp.exists(dir_path) and not force: return dir_path - if not os.path.exists(dir_path): - os.makedirs(dir_path) print('download_path:', dir_path) zip_file_path = os.path.join(_root, sub_dir, name + '.zip') model_url = "%s/%s/%s.zip"%(BASE_REPO_URL, sub_dir, name) download_file(model_url, path=zip_file_path, overwrite=True) + if not os.path.exists(dir_path): + os.makedirs(dir_path) with zipfile.ZipFile(zip_file_path) as zf: zf.extractall(dir_path) os.remove(zip_file_path) return dir_path def ensure_available(sub_dir, name, root='~/.insightface'): - _root = os.path.expanduser(root) - dir_path = os.path.join(_root, sub_dir, name) - if osp.exists(dir_path): - return dir_path return download(sub_dir, name, force=False, root=root)