3j SSKJr SSKrSSKrSSKrSSKJrJr SSKJ r J r J r J r J r JrJr SSKrSSKJr SSKJr SSKJr SSKJr \R2"\5r\ "S 5r"S S 5r"S S \\R<\\ \5rg)) annotationsN)ABCabstractmethod)AnyGenericListOptionalTypeVarUnioncast)nn)PILImage)tensor_to_image) write_image ModelConfigcr\rSrSr%SrSrS\S'S S SjjrS S SjjrSSS jjr S r g)ModelBaseMixin&zAProvide common properties and utilities for Kornia model classes.modelstrnamecUS:XaU$US:Xa<[U[5(aUVs/sHn[U5PM sn$[U5$[SUS35es snf)aConvert the output tensor to the desired type. Args: output: The output tensor or list of tensors. output_type: The desired output type. Accepted values are "torch" and "pil". is_batch: If True, the output is expected to be a batch of tensors. Returns: The converted output tensor or list of tensors. Raises: RuntimeError: If the output type is not supported. torchpilz Output type z9 is not supported. Accepted values are 'torch' and 'pil'.) isinstancelistr RuntimeError)selfoutput output_typeis_batchts L/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/models/base.py_tensor_to_typeModelBaseMixin._tensor_to_type+se" ' !M E !&$''4:;Fq*F;;&v..k]:stu u rD__static_attributes__rFr@r#rr&svKD#chv=vLOv[_v Cv69&oq9=9JW9hk9 99r@rcz\rSrSrSrS S Sjjr\\S Sj55rSSSSSSS.SS jjr S r g) ModelBaseqz2Abstract model class with some utilities function.Nc[RRU5(a*[US5n[R "X2S9nSSS5 O[R RXS9nURW5 g!,(df  N =f)zLoad checkpoint from a given url or file. Args: checkpoint: The url or filepath for the respective checkpoint device: The desired device to load the weights and move the model rb) map_locationN) r/r6isfileopenrloadhubload_state_dict_from_urlload_state_dict)r checkpointdevicef state_dicts r#load_checkpointModelBase.load_checkpointtsi 77>>* % %j$'1"ZZ? (';;J;\J Z( ('s A>> B c[e)z[Build/load the model. Args: config: The specifications for the model be build/loaded )NotImplementedError)configs r# from_configModelBase.from_configs "!r@Finductor fullgraphdynamicbackendmodeoptionsdisablec b[R"XX#XEUS9n[[[U5nU$)Nrl)rcompiler rUr)rrmrnrorprqrrcompileds r#rtModelBase.compiles4== wdmt  +.9r@rF)N)r`rrazOptional[torch.device]rIrJ)rhrrIModelBase[ModelConfig])rmrHrnrHrorrprLrqzOptional[dict[Any, Any]]rrrHrIrw) rMrNrOrPrQrd staticmethodrrirtrSrFr@r#rUrUqs<) "" !",0     * r@rU) __future__rr1loggingr/abcrrtypingrrrr r r r rr kornia.core.externalrImagekornia.image.utilsr kornia.ior getLoggerrMr8rrModulerUrFr@r#rsp$# #EEE 2.!   8 $m$ H9H9V,RYY 0D,r@