,}hF' 8ddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZddlmZd d lmZgd ZeGd d ZGddeZde defdZ!deee fdeefdZ"dedeefdZ#edejHZ%iZ&d!dee deede%fgede%fffdZ' d"dee deee e dfdeee e dfdee fdZ(de dedejHffdZ)de dedejHfd Z*y)#N) dataclass)Enum)partial) signature) ModuleType) AnyCallableDictIterableListMappingOptionalSetTypeTypeVarUnion)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) isinstancerNotImplementedrr rrfuncargskeywords)selfr!s V/var/www/html/test/engine/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 r rboolr+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. objr"c|qt|tur$||j|jdzd}|St ||s0t d|jd|j jd|S)N.z(Invalid Weight class provided; expected z but received )typer1replacer-r$ TypeError __class__)clsr6s r*verifyzWeightsEnum.verifyNs} ?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_dictYs'B4B6BBr,cL|jjd|jS)Nr8)r=r-_name_r)s r*__repr__zWeightsEnum.__repr__\s"..))*!DKK=99r,c.|jjSrB)valuerrFs r*rzWeightsEnum.url_szz~~r,c.|jjSrB)rIrrFs r*rzWeightsEnum.transformscszz$$$r,c.|jjSrB)rIr rFs r*r zWeightsEnum.metagszzr,N)r-r.r/r0 classmethodrr?r r1rCrGpropertyrrr r4r,r*rrDsCCC3C738;LC:#:%%r,rnamer"cn |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. r8zInvalid 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) rN enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs r*rrlsBE $ 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$r1r_get_enum_from_fn)rNmodels r*rrs%(2$'< d #$E U ##r,fncbt|}d|jvr tdt|jdj}d}t |t rt |tr|}n6|jD]'}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 parametersrT annotationr$r:rar__args__)rnsigannrhts r*rlrls B-C&HII B- " "9 - 8 8CL#tC!=  A!T"z!['A    x   r,M)bound.cPdtdtfdtdtfffd }|S)Nrn.r"chn |j}|tvrtd|d|t|<|S)Nz/An entry is already registered under the name 'rP)r-BUILTIN_MODELSrT)rnkeyrNs r*wrapperzregister_model..wrappers@&dBKK . NseSUVW W s r,)r rw)rNr}s` 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. r8rRr) r{itemsr.rsplitr-setr$r1fnmatchfiltersorted) rrrkv 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 )lowerr{KeyErrorrT)rNrns 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. r4)r)rNrrns r*rr s 4 B <<r,rB)NNN)+rrXrZrV dataclassesrenumr functoolsrrtypesrtypingrr r r r r rrrrrtorchr_internally_replaced_utilsr__all__rrr1rrrlModulerwr{r~rrrr4r,r*rs !ddddA y ,7,7 ,7^%$%P!$S!$[!$H $E(C-0 $T+5F $(tK'8B Cryy!#(HS!V