o oi@sbddlmZddlZddlZddlZddlmZmZddl m Z dgZ e e ZGdddZdS)) annotationsN)AnyOptional) kornia_configCachedDownloaderc@sBeZdZdZeddd d ZeddddZe ddddZdS)rzFDownloads files from URLs to the local cache or .kornia_hub directory.N model_namestr cache_dir Optional[str]suffixreturncCs|durtj}|dur||stj|d|}ntj|d}tjjg|tj|tjdd|R}|S)a;Construct the file path for the ONNX model based on the model name and cache directory. Args: model_name: The name of the model or operator, typically in the format 'operators/model_name'. cache_dir: The directory where the model should be cached. Defaults to None, which will use a default `kornia.config.hub_onnx_dir` directory. suffix: Optional file suffix when the filename is the model name. Returns: str: The full local path where the model should be stored or loaded from. N)r hub_cache_direndswithospathsplitjoinsep)clsrr r file_name file_pathrI/home/ubuntu/.local/lib/python3.10/site-packages/kornia/utils/download.py_get_file_path#s2zCachedDownloader._get_file_pathTurlnamedownloadboolkwargsrcKsT|dr#|dd}|dd}|j|||d}|j|||d|Std|)Nzhttp:zhttps:r r )r )download_if_not_existsz-URL must start with 'http:' or 'https:'. Got ) startswithgetrr ValueError)rrrrrr r rrrrdownload_to_cache=s   z"CachedDownloader.download_to_cacherr!Nonec Cstj|rtd|d|ddS|std|dtjtj|dd|d r\ztd |d |dt j ||WdSt j j y[}z td |d|d}~wwtd )aaDownload an ONNX model from the specified URL and save it to the specified file path. Args: url: The URL of the ONNX model to download. file_path: The local path where the downloaded model should be saved. download_if_not_exists: If True, the file will be downloaded if it's not already downloaded. z Loading `z` from `z`.N`z)` not found. You may set `download=True`.T)exist_okr z Downloading `z` to `zError in resolving `z'URL must start with 'http:' or 'https:')rrexistsloggerinfor$makedirsdirnamer"urllibrequest urlretrieveerror HTTPError)rrrr!errrrGs  zCachedDownloader.download)N)rrr r r r r r)T) rrrrrrrrr r)rrrrr!rr r&)__name__ __module__ __qualname____doc__ classmethodrr%rrrrrr s ) __future__rloggingrurllib.requestr.typingrr kornia.configr__all__ getLoggerr4r*rrrrrs