o )iF' @sddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZddlmZd d lmZgd ZeGd d d ZGdddeZde defddZ!deee fdeefddZ"dedeefddZ#edej$dZ%iZ&d(dee deede%fgede%fffddZ'   d)dee deee e dfd eee e dfdee fd!d"Z(de dedej$ffd#d$Z)de d%edej$fd&d'Z*dS)*N) dataclass)Enum)partial) signature) ModuleType) AnyCallableDictIterableListMappingOptionalSetTypeTypeVarUnion)nn)load_state_dict_from_url) WeightsEnumWeights get_modelget_model_builderget_model_weights get_weight list_modelsc@sDeZdZUdZeed<eed<eeefed<dede fddZ d S) ra This class is used to group important attributes associated with the pre-trained weights. Args: url (str): The location where we find the weights. transforms (Callable): A callable that constructs the preprocessing method (or validation preset transforms) needed to use the model. The reason we attach a constructor method rather than an already constructed object is because the specific object might have memory and thus we want to delay initialization until needed. meta (Dict[str, Any]): Stores meta-data related to the weights of the model and its configuration. These can be informative attributes (for example the number of parameters/flops, recipe link/methods used in training etc), configuration parameters (for example the `num_classes`) needed to construct the model or important meta-data (for example the `classes` of a classification model) needed to use the model. url transformsmetaotherreturncCst|tstS|j|jkrdS|j|jkrdSt|jtr;t|jtr;|jj|jjko:|jj|jjko:|jj |jj kS|j|jkS)NF) isinstancerNotImplementedrrrrfuncargskeywords)selfrr'f/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/models/_api.py__eq__)s    zWeights.__eq__N) __name__ __module__ __qualname____doc__str__annotations__rr rboolr)r'r'r'r(rs rc@sveZdZdZededefddZdededeeeffdd Z defd d Z e d d Z e ddZ e ddZdS)raM This class is the parent class of all model weights. Each model building method receives an optional `weights` parameter with its associated pre-trained weights. It inherits from `Enum` and its values should be of type `Weights`. Args: value (Weights): The data class entry with the weight information. objr cCsX|dur*t|tur|||jdd}|St||s*td|jd|jjd|S)N.z(Invalid Weight class provided; expected z but received )typer.replacer*r! TypeError __class__)clsr1r'r'r(verifyNs  zWeightsEnum.verifyr$kwargscOst|jg|Ri|SN)rr)r&r$r:r'r'r(get_state_dictYszWeightsEnum.get_state_dictcCs|jjd|jS)Nr2)r7r*_name_r&r'r'r(__repr__\szWeightsEnum.__repr__cC|jjSr;)valuerr>r'r'r(r_zWeightsEnum.urlcCr@r;)rArr>r'r'r(rcrBzWeightsEnum.transformscCr@r;)rArr>r'r'r(rgrBzWeightsEnum.metaN)r*r+r,r- classmethodrr9r r.r<r?propertyrrrr'r'r'r(rDs    rnamer c Csz |d\}}Wntytd|dwdtjtjddd}t|}|gddt |tj D}d}|D]}|j |d}|durWt |trW|}nqA|durdtd|d ||S) z Gets the weights enum value by its full name. Example: "ResNet50_Weights.IMAGENET1K_V1" Args: name (str): The name of the weight enum entry. Returns: WeightsEnum: The requested weight enum. r2zInvalid weight name provided: ''.NcSs$g|]}|djdr|dqS)z __init__.py)__file__endswith).0xr'r'r( }s zget_weight..zThe weight enum 'z0' for the specific method couldn't be retrieved.)split ValueErrorjoinsysmodulesr* importlib import_moduleinspect getmembersismodule__dict__get issubclassr) rEZ enum_nameZ value_nameZbase_module_name base_moduleZ model_modules weights_enummZpotential_classr'r'r(rls(      rcCst|tr t|n|}t|S)a Returns the weights enum class associated to the given model. Args: name (callable or str): The model builder function or the name under which it is registered. Returns: weights_enum (WeightsEnum): The weights enum class associated with the model. )r!r.r_get_enum_from_fn)rEmodelr'r'r(rs rfncCst|}d|jvr tdt|jdj}d}t|tr$t|tr$|}n|jD]}t|tr7t|tr7|}nq'|dur@td|S)z Internal method that gets the weight enum of a specific model builder method. Args: fn (Callable): The builder method used to create the model. Returns: WeightsEnum: The requested weight enum. weightsz-The method is missing the 'weights' argument.NzjThe WeightsEnum class for the specific method couldn't be retrieved. Make sure the typing info is correct.) r parametersrO annotationr!r4rZr__args__)r`sigannr\tr'r'r(r^s"  r^M)bound.cs*dtdtfdtdtfffdd }|S)Nr`.r cs6durn|j}|tvrtd|d|t|<|S)Nz/An entry is already registered under the name 'rF)r*BUILTIN_MODELSrO)r`keyrEr'r(wrappers zregister_model..wrapper)rrh)rErmr'rlr(register_models&rnmoduleincludeexcludecsfddtD}|r(t}t|tr|g}|D] }|tt||B}qn|}|rCt|tr4|g}|D] }|tt||}q6t|S)a Returns a list with the names of registered models. Args: module (ModuleType, optional): The module from which we want to extract the available models. include (str or Iterable[str], optional): Filter(s) for including the models from the set of all models. Filters are passed to `fnmatch `__ to match Unix shell-style wildcards. In case of many filters, the results is the union of individual filters. exclude (str or Iterable[str], optional): Filter(s) applied after include_filters to remove models. Filter are passed to `fnmatch `__ to match Unix shell-style wildcards. In case of many filters, the results is removal of all the models that match any individual filter. Returns: models (list): A list with the names of available models. cs4h|]\}}dus|jdddjkr|qS)Nr2rHr)r+rsplitr*)rKkvror'r( s.zlist_models..)rjitemssetr!r.fnmatchfiltersorted)rorprqZ all_modelsmodelsZinclude_filterZexclude_filterr'rur(rs"   rcCs4|}zt|}W|Stytd|w)z Gets the model name and returns the model builder method. Args: name (str): The name under which the model is registered. Returns: fn (Callable): The model builder method. zUnknown model )lowerrjKeyErrorrO)rEr`r'r'r(rs   rconfigcKst|}|di|S)a& Gets the model name and configuration and returns an instantiated model. Args: name (str): The name under which the model is registered. **config (Any): parameters passed to the model builder method. Returns: model (nn.Module): The initialized model. Nr')r)rErr`r'r'r(r s rr;)NNN)+ryrSrUrQ dataclassesrenumr functoolsrrtypesrtypingrrr r r r r rrrrZtorchrZ_internally_replaced_utilsr__all__rrr.rrr^Modulerhrjrnrrrr'r'r'r(sH     4  /($!2  (