Yh& ddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddlm Z ddlmZddlmZmZmZmZmZmZdd lmZd d lmZgd ZeGd dZGdde ZdedefdZdeeefde efdZ!dede efdZ"edej#Z$iZ%d"deedeede$fgede$fffdZ& d#deedeeeedfdeeeedfde'efdZ(dededej#ffdZ)ded edej#fd!Z*dS)$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_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) isinstancerNotImplementedrrrrfuncargskeywords)selfrs i/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/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 dictr boolr(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. objrc|pt|tur%|||jdzd}n5t ||s%t d|jd|jjd|S)N.z(Invalid Weight class provided; expected z but received )typer.replacer*r! TypeError __class__)clsr4s r'verifyzWeightsEnum.verifyOs ?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_dictZs"'B4BBB6BBBr)c.|jjd|jS)Nr6)r;r*_name_r&s r'__repr__zWeightsEnum.__repr__]s.)99DK999r)c|jjSr@)valuerrDs r'rzWeightsEnum.url`s z~r)c|jjSr@)rGrrDs r'rzWeightsEnum.transformsds z$$r)c|jjSr@)rGrrDs r'rzWeightsEnum.metahs zr)N)r*r+r,r- classmethodr r=rr.rArEpropertyrrrr2r)r'rrEs[CCC3C738;LCCCC:#::::X%%X%Xr)rnamercB |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. r6zInvalid 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) rL enum_name value_namebase_module_name base_module model_modules weights_enummpotential_classs r'rrmsKE $ 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!r.r_get_enum_from_fn)rLmodels r'rrs4(2$'<'< F d # # #$E U # ##r)fncxt|}d|jvrtd|jdj}d}t |t rt |tr|}n@t|D]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 parametersrX annotationr!r8rcrr )rosigannrits r'rmrms B--C&&HIII . # .CL#tC!=!= #  A!T"" z!['A'A   x    r)M)bound.cddtdtfdtdtfffd }|S)Nro.rcjn|j}|tvrtd|d|t|<|S)Nz/An entry is already registered under the name 'rN)r*BUILTIN_MODELSrX)rokeyrLs r'wrapperzregister_model..wrappersG&ddBK . VsVVVWW W s r))r rw)rLr}s` r'register_modelr~sHHS!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)Nr6rQr)r+rsplitr*)rTkvrs r' zlist_models..sPa!,BUBUVY[\B]B]^_B`djdsBsBsBsBsBsr))r{itemssetr!r.fnmatchfiltersorted)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{KeyErrorrX)rLros r'rrsY ::<rs --------!!!!!!DDDDDDDDDDDDDDDDAAAAAA y x x ,7,7,7,7,7,7,7 ,7^%%%%%$%%%P!$S!$[!$!$!$!$H $E(C-0 $T+5F $ $ $ $(tK'8@ GCry!!!#(HS!V