j[( ddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZddlmZd d lmZgd ZeGd d ZGddeZde defdZ!deee fdeefdZ"dedeefdZ#edej$Z%iZ&d!dee deede%fgede%fffdZ' d"dee deee e dfdeee e dfdee fdZ(de dedej$ffdZ)de dedej$fd 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_modelscReZdZUdZeed<eed<eeefed<dede fdZ dS) 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 transformsmetaotherreturnct|tstS|j|jkrdS|j|jkrdSt|jt rht|jt rN|jj|jjko3|jj|jjko|jj |jj kS|j|jkS)NF) isinstancerNotImplementedrr rrfuncargskeywords)selfr!s HC:\Projects\Historical\venv\Lib\site-packages\torchvision/models/_api.py__eq__zWeights.__eq__)s%)) "! ! 8uy 5 9 " "5 dow / / 7Ju?OQX4Y4Y 7$(8(==JO(E,<,AAJO,0@0II  ?e&66 6N) __name__ __module__ __qualname____doc__str__annotations__r r rboolr+r,r*rrsm   HHH sCx.7C7D777777r,rceZdZdZededefdZdededeeeffdZ defdZ e d Z e d Z e d Zd S) 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|pt|tur%|||jdzd}n5t ||s%t d|jd|jjd|S)N.z(Invalid Weight class provided; expected z but received )typer1replacer-r$ TypeError __class__)clsr6s r*verifyzWeightsEnum.verifyNs ?CyyC#++clS&8"==>S)) ts|tt[^[h[qttt r,r'kwargsc,t|jg|Ri|SN)rr)r)r'r@s r*get_state_dictzWeightsEnum.get_state_dictYs"'B4BBB6BBBr,c.|jjd|jS)Nr8)r=r-_name_r)s r*__repr__zWeightsEnum.__repr__\s.)99DK999r,c|jjSrB)valuerrFs r*rzWeightsEnum.url_s z~r,c|jjSrB)rIrrFs r*rzWeightsEnum.transformscs z$$r,c|jjSrB)rIr rFs r*r zWeightsEnum.metags zr,N)r-r.r/r0 classmethodrr?r r1rCrGpropertyrrr r4r,r*rrDs[CCC3C738;LCCCC:#::::X%%X%Xr,rnamer"cB |d\}}n!#t$rtd|dwxYwdtjt jddd}t j|}|gdtj |tj Dz}d}|D]8}|j |d}|t|tr|}n9|td|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. r8zInvalid weight name provided: ''.Nc^g|]*}|djd"|d+S)z __init__.py)__file__endswith).0xs r* zget_weight..}sF%%% Q4= ! !- 0 0% !%%%r,zThe weight enum 'z0' for the specific method couldn't be retrieved.)split ValueErrorjoinsysmodulesr- importlib import_moduleinspect getmembersismodule__dict__get issubclassr) rN enum_name value_namebase_module_name base_module model_modules weights_enummpotential_classs r*rrlsKE $ 3 :: EEEC4CCCDDDExx H 5 > D DS I I#2# NOO)*:;;K M%%#K1ABB%%%M L *..D99  &:o{+S+S &*L EhYhhhiii  ##s9clt|trt|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*rrs4(2$'<'< F d # # #$E U # ##r,fnct|}d|jvrtdt|jdj}d}t |t rt |tr|}n8|jD]0}t |t rt |tr|}n1|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 parametersrZ annotationr$r:rer__args__)rqsigannrkts r*roros B--C&&HIII B-- "9 - 8CL#tC!=!=   A!T"" z!['A'A   x    r,M)bound.cddtdtfdtdtfffd }|S)Nrq.r"cjn|j}|tvrtd|d|t|<|S)Nz/An entry is already registered under the name 'rP)r-BUILTIN_MODELSrZ)rqkeyrNs r*wrapperzregister_model..wrappersG&ddBK . VsVVVWW W s r,)r rz)rNrs` r*register_modelrsHHS!V$#q&)9 Nr,moduleincludeexcludecfdtD}|rQt}t|tr|g}|D]'}|tt j||z}(n|}|rBt|tr|g}|D]'}|tt j||z }(t|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. crh|]3\}}*|jdddjk1|4S)Nr8rSr)r.rsplitr-)rVkvrs r* zlist_models..sPa!,BUBUVY[\B]B]^_B`djdsBsBsBsBsBsr,)r~itemssetr$r1fnmatchfiltersorted)rrr all_modelsmodelsinclude_filterexclude_filters` r*rrs($**,,J55 gs # # iG% N NNc'.^"L"LMMMFF NN gs # # iG% N NNc'.^"L"LMMMFF &>>r,c|} t|}n #t$rtd|wxYw|S)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~KeyErrorrZ)rNrqs r*rrsY ::<rs !!!!!!ddddddddddddddddddddddddddAAAAAA y x x ,7,7,7,7,7,7,7 ,7^%%%%%$%%%P!$S!$[!$!$!$!$H $E(C-0 $T+5F $ $ $ $(tK'8B GCry!!!#(HS!V