| |
|
| | import os |
| | import os.path as osp |
| | import zipfile |
| | from .download import download_file |
| |
|
| | BASE_REPO_URL = 'https://github.com/deepinsight/insightface/releases/download/v0.7' |
| |
|
| | def download(sub_dir, name, force=False, root='~/.insightface'): |
| | _root = os.path.expanduser(root) |
| | dir_path = os.path.join(_root, sub_dir, name) |
| | if osp.exists(dir_path) and not force: |
| | return dir_path |
| | print('download_path:', dir_path) |
| | zip_file_path = os.path.join(_root, sub_dir, name + '.zip') |
| | model_url = "%s/%s.zip"%(BASE_REPO_URL, 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) |
| | |
| | return dir_path |
| |
|
| | def ensure_available(sub_dir, name, root='~/.insightface'): |
| | return download(sub_dir, name, force=False, root=root) |
| |
|
| | def download_onnx(sub_dir, model_file, force=False, root='~/.insightface', download_zip=False): |
| | _root = os.path.expanduser(root) |
| | model_root = osp.join(_root, sub_dir) |
| | new_model_file = osp.join(model_root, model_file) |
| | if osp.exists(new_model_file) and not force: |
| | return new_model_file |
| | if not osp.exists(model_root): |
| | os.makedirs(model_root) |
| | print('download_path:', new_model_file) |
| | if not download_zip: |
| | model_url = "%s/%s"%(BASE_REPO_URL, model_file) |
| | download_file(model_url, |
| | path=new_model_file, |
| | overwrite=True) |
| | else: |
| | model_url = "%s/%s.zip"%(BASE_REPO_URL, model_file) |
| | zip_file_path = new_model_file+".zip" |
| | download_file(model_url, |
| | path=zip_file_path, |
| | overwrite=True) |
| | with zipfile.ZipFile(zip_file_path) as zf: |
| | zf.extractall(model_root) |
| | return new_model_file |
| |
|
