3j%SSKJr SSKrSSKrSSKrSSKrSSKJrJr SSK r SSK J r SSK J r SSK JrJr SSKJr /SQr\R("\5r"S S \5rSS jrSSS jjrSS jrg)) annotationsN)AnyOptional) kornia_config)numpy)onnxrequests)CachedDownloader) ONNXLoader add_metadataio_name_conversionc\rSrSrSr\S S Sjj5r\S S Sjj5r\SSj5r \SSj5r \SSj5r Sr g )r %z_Manages ONNX models, handling local caching, downloading from Hugging Face, and loading models.c bURS5(adUR"U[RR U5S4USS.UD6n[ U5n[ R"U5nUsSSS5 $U(d[SUS35e[SUS35e!,(df  N3=f) a,Load JSON config from the specified URL. Args: url: The URL of the preprocessor config to load. download: If True, the config will be downloaded if it's not already in the local cache. kwargs: Additional download arguments. Returns: dict[str, Any]: The loaded preprocessor config. )zhttp:zhttps:z.jsondownloadsuffixNzFile `z)` not found. You may set `download=True`.z ` not found.) startswithdownload_to_cacheospathsplitopenjsonload RuntimeError)clsurlrkwargs file_pathf json_datas K/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/onnx/utils.py load_configONNXLoader.load_config(s >>- . .-- c"2&"   IiA IIaL  !u,UVW WVC5 566!s B  B.c URS5(aU[S5SnSUS3nURSS5=(d@ [RR [ RURS5S5nURSU05 UR"XQRS5S4USS .UD6nU(a.SUS 3nUR"XRS5S4US S .UD6 [R"U5$URS 5(dURS 5(aURSS5=(d [ RnURSU05 UR"U[RRU5S 4USS .UD6nU(aEUSSS -nUR"U[RRU5S SSS -4US S .UD6 [R"U5$[RRU5(a[R"U5$[SUS35e)a:Load an ONNX model from the local cache or downloads it from Hugging Face if necessary. Args: model_name: The name of the ONNX model or operator. For Hugging Face-hosted models, use the format 'hf://model_name'. Valid `model_name` can be found on https://huggingface.co/kornia/ONNX_models. Or a URL to the ONNX model. download: If True, the model will be downloaded from Hugging Face if it's not already in the local cache. cache_dir: The directory where the model should be cached. Defaults to None, which will use a default `{kornia_config.hub_onnx_dir}` directory. with_data: If True, the model will be loaded with its `.onnx_data` weights. **kwargs: Additional arguments to pass to the download method, if needed. Returns: onnx.ModelProto: The loaded ONNX model. zhf://Nz7https://huggingface.co/kornia/ONNX_models/resolve/main/z.onnx cache_dir/rrz .onnx_datazhttp://zhttps://rzFile z not found)rlengetrrjoinr hub_onnx_dirrupdaterrrexists ValueError) r model_namer with_datar rr(r!url_datas r$ load_modelONNXLoader.load_modelFsG&   ) )#CLN3JKJrQrSzlist[dict[str, Any]])rSNone) __name__ __module__ __qualname____firstlineno____doc__ classmethodr%r6 staticmethodr@rIrN__static_attributes__rPr8r$r r %sji77:;9;9z]] r8r c[[URR55HMnURRURnX1;dM-XURRUlMO [[URR 55HMnURR URnXA;dM-XURR UlMO [[URR 55GH&n[[URR UR55HXnURR URUnXa;dM0XURR URU'MZ [[URR UR 55HXnURR UR UnXq;dM0XURR UR U'MZ GM) U$)zConvert the input and output names of an ONNX model to 'input' and 'output'. Args: onnx_model: The ONNX model to convert. io_name_mapping: A dictionary mapping the original input and output names to the new ones. )ranger,graphinputnameoutputnode) onnx_modelio_name_mappingiin_nameout_namej node_in_name node_out_names r$r r s3z''--. /""((+00  %-<-EJ   " "1 % *0 3z''../ 0##**1-22  &.=.GJ   # #A & +1 3z'',,- .s:++00399:;A%++00399!M/5D5S   %%a(//2= / r8cUc/nSS[R4/UQH5up#URR5nX$l[ U5UlM7 U$)aAdd metadata to an ONNX model. The metadata includes the source library (set to "kornia"), the version of kornia, and any additional metadata provided as a list of key-value pairs. Args: onnx_model: The ONNX model to add metadata to. additional_metadata: A list of tuples, where each tuple contains a key and a value for the additional metadata to add to the ONNX model. Returns: The ONNX model with the added metadata. )sourcekorniaversion)ro __version__metadata_propsaddkeyrQvalue)readditional_metadatartrurrs r$r r si$"  F&&'   $22668 "5z r8c ([R[R[R[R[R [R [R[RS.nX;a[SUS35eX$)N)z tensor(float)ztensor(float16)ztensor(double)z tensor(int32)z tensor(int64)z tensor(uint8)z tensor(int8)z tensor(bool)z ONNX type z not understood) npfloat32float16float64int32int64uint8int8bool_ TypeError) onnx_type type_mappings r$onnx_type_to_numpyrsf::** L$*YK?@@  ""r8)rerTrfzdict[str, str]rSrT)N)rerTrvzOptional[list[tuple[str, str]]]rSrT)rrQrSr) __future__rrloggingrrEtypingrrro kornia.configrkornia.core.externalrrxrr kornia.onnx.downloadr __all__ getLoggerrVloggerr r r rrPr8r$rs$#  ',/1 >   8 $B!BJ""#""N<@8> #r8