insightface icon indicating copy to clipboard operation
insightface copied to clipboard

utils/storage.py error

Open MATEMATNKx opened this issue 1 year ago • 0 comments

https://github.com/deepinsight/insightface/blob/v0.7/python-package/insightface/utils/storage.py#L13

Error in line for Windows here

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/%s.zip"%(BASE_REPO_URL, sub_dir, name)
    #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)
    #os.remove(zip_file_path)
    return dir_path

def ensure_available(sub_dir, name, root='~/.insightface'):
    return download(sub_dir, name, force=False, root=root)

The issue https://github.com/deepinsight/insightface/issues/1525#issuecomment-2510257429

You need change code to:

def download(sub_dir, name, force=False, root='~/.insightface'):
   _root = os.path.expanduser(root)
   extract_path = os.path.join(_root, sub_dir)
   dir_path = os.path.join(extract_path, 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(extract_path)
   #os.remove(zip_file_path)
   return dir_path
   ```
   with care, MATEMATNK

MATEMATNKx avatar Dec 01 '24 22:12 MATEMATNKx