o oi @s`ddlmZddlZddlZddlmZddlmZddlm Z m Z m Z     ddddZ dS)) annotationsN)Optional)urlparse) HASH_REGEXdownload_url_to_fileget_dirTFurlstr model_dir Optional[str]progressbool check_hash file_namereturnc Cs|durt}tj|d}tj|ddt|}tj|j}|dur&|}tj||}tj|sZtj d|d|dd} |rRt |} | rP| dnd} t||| |d |S) aLoad the ONNX model at the given URL. If downloaded file is a zip file, it will be automatically decompressed. If the object is already present in `model_dir`, it's deserialized and returned. The default value of ``model_dir`` is ``/checkpoints`` where ``hub_dir`` is the directory returned by :func:`~torch.hub.get_dir`. Args: url (str): URL of the object to download model_dir (str, optional): directory in which to save the object progress (bool, optional): whether or not to display a progress bar to stderr. Default: True check_hash(bool, optional): If True, the filename part of the URL should follow the naming convention ``filename-.ext`` where ```` is the first eight or more digits of the SHA256 hash of the contents of the file. The hash is used to ensure unique names and to verify the contents of the file. Default: False file_name (str, optional): name for the downloaded file. Filename from ``url`` will be used if not set. Example: >>> model = download_onnx_from_url('https://github.com/fabio-sim/LightGlue-ONNX/releases/download/v1.0.0/disk_lightglue_fused_fp16.onnx') N checkpointsT)exist_okzDownloading: "z" to  )r )rospathjoinmakedirsrbasenameexistssysstderrwritersearchgroupr) rr r rrhub_dirpartsfilename cached_file hash_prefixrr&`/home/ubuntu/.local/lib/python3.10/site-packages/kornia/feature/lightglue_onnx/utils/download.pydownload_onnx_from_urls"!  r()NTFN) rr r r r r rr rr rr ) __future__rrrtypingr urllib.parser torch.hubrrrr(r&r&r&r's