h& @SSKrSSKrSSKrSSKrSSKJrJr SSKJr SSK J r SSK J r SSKJ r SSKJr SSKJrJrJrJrJrJr SS KJr S S KJr /S Qr\"S S55r"SS\ 5rS\S\4SjrS\\\4S\ \4Sjr!S\S\ \4Sjr"\"S\RFS9r$0r%S"S\\S\\S\$4/\S\$444Sjjr&S#S\\S\\\\S4S\\\\S4S\'\4Sjjr(S\S\S\RF44Sjr)S\S \S\RF4S!jr*g)$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_modelscT\rSrSr%Sr\\S'\\S'\\\ 4\S'S\ S\ 4Sjr S r g ) 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 transformsmetaotherreturnc|[U[5(d[$URUR:wagURUR:wag[UR [ 5(a[UR [ 5(aUR RUR R:H=(aa UR RUR R:H=(a- UR RUR R:H$UR UR :H$)NF) isinstancerNotImplementedrrrrfuncargskeywords)selfrs O/var/www/auris/envauris/lib/python3.13/site-packages/torchvision/models/_api.py__eq__Weights.__eq__*s%))! ! 88uyy  99 " doow / /Ju?O?OQX4Y4Y$$(8(8(=(==JOO((E,<,<,A,AAJOO,,0@0@0I0II  ??e&6&66 6N) __name__ __module__ __qualname____firstlineno____doc__str__annotations__r dictr boolr)__static_attributes__r,r+r(rrs5  H sCx.7C7D7r+rc\rSrSrSr\S\S\4Sj5rS\S\S\\ \44Sjr S\ 4S jr \ S 5r \ S 5r\ S 5rS rg)rEa5 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 cUbu[U5[La#XRURS-S5nU$[ X5(d0[ SURSUR RS35eU$)N.z(Invalid Weight class provided; expected z but received )typer2replacer-r" TypeError __class__)clsr9s r(verifyWeightsEnum.verifyOs| ?CyC++cllS&8"=>   ))>s||nN[^[h[h[q[qZrrst r+r%kwargsc4[UR/UQ70UD6$N)rr)r'r%rDs r(get_state_dictWeightsEnum.get_state_dictZs'B4B6BBr+cLURRSUR3$)Nr;)r@r-_name_r's r(__repr__WeightsEnum.__repr__]s"..))*!DKK=99r+c.URR$rF)valuerrKs r(rWeightsEnum.url`szz~~r+c.URR$rF)rOrrKs r(rWeightsEnum.transformsdszz$$$r+c.URR$rF)rOrrKs r(rWeightsEnum.metahszzr+r,N)r-r.r/r0r1 classmethodr rBrr2rGrLpropertyrrrr6r,r+r(rrEsCCC3C738;LC:#:%%r+rnamer cURS5upSR[R[ R RS5SS5n[ R"U5nU/[R"U[R5Vs/sH-nUSRRS5(dM(USPM/ sn-nSnUH=nURRUS5n U cM$[U [ 5(dM;U n O Uc[SUS 35eXr$![a [SUS35ef=fs snf) 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. r;zInvalid 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) rW enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs r(rrmsCE $ 3 xx H 5 > > D DS I#2 NO))*:;K M##K1A1AB%BA Q4== ! !- 0 !B%M L **..D9  &:o{+S+S*L   ,YK7ghii  ##+ E:4&CDDE %sD*'E E*Ec\[U[5(a [U5OUn[U5$)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"r2r_get_enum_from_fn)rWmodels r(rrs((2$'<'< d #$E U ##r+fnc[U5nSUR;a [S5eURSRnSn[ U[ 5(a[ U[5(aUnOB[U5H3n[ U[ 5(dM[ U[5(dM1Un O Uc [S5eU$)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 parametersr] annotationr"r=rjrr )rwsigannrqts r(rurus B-C&HII .. # . .CL#tC!=!= #A!T""z!['A'A   x   r+M)bound.cT^S[S[4S[S[44U4SjjnU$)Nrw.r cn>TbTO URnU[;a[SUS35eU[U'U$)Nz/An entry is already registered under the name 'rY)r-BUILTIN_MODELSr])rwkeyrWs r(wrapperregister_model..wrappers@&dBKK . NseSUVW W s r+)r r)rWrs` r(register_modelrs/HS!V$#q&)9 Nr+moduleincludeexcludec[R5VVs1sH9up4Ub/URRSS5SUR:XdM7UiM; nnnU(aN[ 5n[ U[5(aU/nUH%nU[ [R"XW55-nM' OUnU(aC[ U[5(aU/nUH%nU[ [R"XX55- nM' [U5$s snnf)aw 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. r;r[r) ritemsr.rsplitr-setr"r2fnmatchfiltersorted) rrrkv all_modelsmodelsinclude_filterexclude_filters r(rrs*%**,,da!,,BUBUVY[\B]^_B`djdsdsBs,5 gs # #iG%Nc'.."LMMF& gs # #iG%Nc'.."LMMF& &>#s 6DDcrUR5n[UnU$![a [SU35ef=f)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 )lowerrKeyErrorr])rWrws r(rrsH ::$0112s 6configc ([U5nU"S0UD6$)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. r,)r)rWrrws r(rr s 4 B <<r+rF)NNN)+rrarcr_collections.abcrr dataclassesrenumr functoolsrrtypesr typingr r r r rrtorchr_internally_replaced_utilsr__all__rrr2rr=rruModulerrrlistrrrr,r+r(rs -!DDA y ,7,7 ,7^%$%P!$S!$[!$H $E(C-0 $T+5F $(tK'8@ Cryy!#(HS!V