:j& 4ddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddlm Z ddlmZddlmZmZmZmZmZmZdd lmZd d lmZgd ZeGd dZGdde ZdedefdZdeeefde efdZ!dede efdZ"edejFZ$iZ%d"deedeede$fgede$fffdZ& d#deedeeeedfdeeeedfde'efdZ(dededejFffdZ)ded edejFfd!Z*y)$N)IterableMapping) dataclass)Enum)partial) signature) ModuleType)AnyCallableget_argsOptionalTypeVarUnion)nn)load_state_dict_from_url) WeightsEnumWeights get_modelget_model_builderget_model_weights get_weight list_modelscJeZdZUdZeed<eed<eeefed<dede fdZ y) 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 transformsmetaotherreturncFt|tstS|j|jk7ry|j|jk7ryt|j t rt|j t r|j j|j jk(xr\|j j|j jk(xr-|j j|j jk(S|j |j k(S)NF) isinstancerNotImplementedrrrrfuncargskeywords)selfrs l/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchvision/models/_api.py__eq__zWeights.__eq__*s%)! ! 88uyy  99 " doow /Ju?O?OQX4Y$$(8(8(=(==JOO((E,<,<,A,AAJOO,,0@0@0I0II  ??e&6&66 6N) __name__ __module__ __qualname____doc__str__annotations__r dictr boolr(r)r'rrs5  H sCx.7C7D7r)rceZdZdZededefdZdededeeeffdZ defdZ e d Z e d Z e d Zy ) 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. objrc|qt|tur$||j|jdzd}|St ||s0t d|jd|j jd|S)N.z(Invalid Weight class provided; expected z but received )typer.replacer*r! TypeError __class__)clsr4s r'verifyzWeightsEnum.verifyOs} ?CyC#++cllS&8"=>   S)>s||nN[^[h[h[q[qZrrst r)r$kwargsc4t|jg|i|SN)rr)r&r$r>s r'get_state_dictzWeightsEnum.get_state_dictZs'B4B6BBr)cL|jjd|jS)Nr6)r;r*_name_r&s r'__repr__zWeightsEnum.__repr__]s"..))*!DKK=99r)c.|jjSr@)valuerrDs r'rzWeightsEnum.url`szz~~r)c.|jjSr@)rGrrDs r'rzWeightsEnum.transformsdszz$$$r)c.|jjSr@)rGrrDs r'rzWeightsEnum.metahszzr)N)r*r+r,r- classmethodr r=rr.rArEpropertyrrrr2r)r'rrEsCCC3C738;LC:#:%%r)rnamercn |jd\}}djtjt j jddd}t j|}|gtj|tjDcgc]%}|djjdr|d'c}z}d}|D]5}|jj|d} | "t| t s3| }n|td|d ||S#t$rtd|dwxYwcc}w) 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. r6zInvalid weight name provided: ''.Nz __init__.pyzThe weight enum 'z0' for the specific method couldn't be retrieved.)split ValueErrorjoinsysmodulesr* importlib import_moduleinspect getmembersismodule__file__endswith__dict__get issubclassr) rL enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs r'rrmsBE $ 3 :xx H 5 > > D DS I#2 NO))*:;K M##K1A1AB% Q4== ! !- 0 !%M L **..D9  &:o{+S*L   ,YK7ghii  ##+ E:4&CDDE %sD*D2D/cRt|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)rLmodels r'rrs%(2$'< d #$E U ##r)fncNt|}d|jvr td|jdj}d}t |t rt |tr|}n5t|D]'}t |t st |ts%|}n| 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 parametersrR annotationr!r8r_rr )rlsigannrfts r'rjrjs B-C&HII .. # . .CL#tC!= # A!T"z!['A    x   r)M)bound.cPdtdtfdtdtfffd }|S)Nrl.rchn |j}|tvrtd|d|t|<|S)Nz/An entry is already registered under the name 'rN)r*BUILTIN_MODELSrR)rlkeyrLs r'wrapperzregister_model..wrappers@&dBKK . NseSUVW W s r))r rt)rLrzs` r'register_modelr{s/HS!V$#q&)9 Nr)moduleincludeexcludectjDchc]6\}}|-|jjddd|jk(s5|8}}}|rGt }t |tr|g}|D]$}|t tj||z}&n|}|r`__ 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. r6rPr) rxitemsr+rsplitr*setr!r.fnmatchfiltersorted) r|r}r~kv all_modelsmodelsinclude_filterexclude_filters r'rrs*%**,a!,,BUBUVY[\B]^_B`djdsdsBsJ5 gs #iG% NNc'..^"LMMF N gs #iG% NNc'..^"LMMF N &>#s 6C*C*cp|j} t|}|S#t$rtd|wxYw)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 )lowerrxKeyErrorrR)rLrls r'rrsH ::$0112s 5configc (t|}|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. r2)r)rLrrls r'rr s 4 B <<r)r@)NNN)+rrVrXrTcollections.abcrr dataclassesrenumr functoolsrrtypesr typingr r r r rrtorchr_internally_replaced_utilsr__all__rrr.rr8rrjModulertrxr{listrrrr2r)r'rs -!DDA y ,7,7 ,7^%$%P!$S!$[!$H $E(C-0 $T+5F $(tK'8@ Cryy!#(HS!V